【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设备名字