【2020年7月】anaconda+pytorch-gpu

【2020年7月】anaconda+pytorch-gpu

之前我已经在anaconda下装了tensorflow-gpu，结果组里的老师建议装pytorch，于是又折腾一天多。不过，这次又经验，比较快。

一、在anaconda下配置环境

可以用命令conda create，也可以用anaconda navigator

二、下载whl文件

因为清华软件园没有相应的资源了，官网用conda install 安装很慢，几个小时过去了还失败了，所以把文件下到本地安装。先上官网](https://pytorch.org/get-started/locally/)相应的文件
选项都是选择的，下载的文件用pip安装，所以选了pip。留意最下面的一行，基本命令可以参见这个。如果是官网用命令下载，直接把这个命名输入即可。
不过速度不行，所以可以轮子下载网址下下来上面对应的两个文件，下的时候留意python版本和处理器型号。

三、安装

激活环境，进入 文件所在路径，调用python,采用 pip install+文件名。

四、测试是否成功

import torch
import torchvision

另外
print(torch.version)
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.current_device())
print(torch.cuda.get_device_name())
分别查看pytorch版本，看cuda是否可用，设备数量，当前cuda设备索引，从0开始，gpu设备名字