美文网首页程序员
ML Note1 使用Conda安装PyTorch

ML Note1 使用Conda安装PyTorch

作者: Ryker的3D龙 | 来源:发表于2020-05-03 02:24 被阅读0次

    目的

    记录实践过程,便于日后回顾,哪天环境重装了还得翻来看

    步骤

    1. 安装pytorch到编译环境
    • 启动cmd命令行
    • 查看conda环境列表
    conda env list
    
    • 添加项目环境,环境名为pytorch,python版本3.7
    conda create -n pytorch python=3.7
    
    • 切换环境
    activate pytorch
    
    • 在该环境中安装pytoch,cuda版本为10.1(GPU加速)
    conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
    

    此代码可从 https://pytorch.org/get-started/locally/ 按需取得,下载速度可观

    1. pycharm中使用此conda环境解释器
    • File-Setting-Project Interperter
      添加Conda环境 D:\Anaconda3\envs\pytorch\python.exe

    相关文章

      网友评论

        本文标题:ML Note1 使用Conda安装PyTorch

        本文链接:https://www.haomeiwen.com/subject/eeerghtx.html