安装pytorch-gpu版本(分享我的成功案例)

用官网的命令太慢,换国内的源也不行,因为国内的源会默认给你安装cpu版本,所以,,,还是往下看吧

一、如果还没安装Anaconda和Cuda的先去这个链接(https://blog.csdn.net/LuoQingShan_lqs/article/details/105376143)安装前三步后再来安装Pytorch。

二:创建一个独有的虚拟环境

     1、创建 conda 环境

           启动 Anaconda prompt 

conda create -n xxxx pip python=3.6    #xxxx是你虚拟环境的名,想要啥就啥,python版本号可换

     2、激活创建的 conda 环境,进入创建的虚拟环境

            conda activate xxxx

二、开始安装pytorch-gpu

  • 打开清华源下的pytorch库
  • https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
  • 找到自己合适的版本,比如我需要GPU版本的并且支持cuda10,网页ctrl+f快捷搜索,找到文件中
 - pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2

       版本号:1.2.0    ------根据自己需求更改

  • nightly版本可以看作是开发版,一般不选用,
    命名格式为
     pytorch-版本-python版本-cuda表示gpu版-cudnn版本.tar.bz2大小一般为5、600M

  • 下载后,放到合适的盘,接着在刚才创建的那个虚拟环境下,cd到文件目录
    输入命令

  • conda install --offline pytorch-1.2.0-py3.7_cuda100_cudnn7_1.tar.bz2 #文件全称
    

    安装完pytorch后,直接输入命令安装完依赖的包:

  • conda install torchvision
    

    打开conda中的python进行测试

  • import torch
    print(torch.cuda.is_available())
    print(torch.version.cuda)
    

    输入结果为True和10.0就对了

你可能感兴趣的:(默认分类)