1.安装pytorch:
我是用pip去安装的,我的python版本是3.7,注意在cuda那里一定不能选none,我直接选的是10.1版本的pytorch.cuda,将下面的那段command运行一下
中途你会发现安装的速度很慢,那么你就直接到:https://download.pytorch.org/whl/torch_stable.html这个地址下载相关的轮子,具体的轮子幸好会出现在你一开始用那段cmd命令下,下载后直接使用
pip install xxx.whl
2.测试pytorch和它的cuda的安装版本和是否能够调用:
import torch
print(torch.__version__ )
print(torch.version.cuda)
print(torch.cuda.is_available())
此时显式出来的结果是:
1.3.1
10.1
False
说明torch可以使用,但是最后的False说明使用cuda不成功
3.