赞
踩
安装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)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。