问题:pip install torch报错
ModuleNotFoundError: No module named 'tools.nnwrap'
error: [Errno 2] No such file or directory: ‘.gitignore’
ModuleNotFoundError: No module named ‘tools.nnwrap’
解决办法:到官网https://pytorch.org/根据自己的版本选择后复制链接
问题:直接复制回车还是会报错
解决办法:记得把“==”前后的空格给删掉
问题:下载速度过慢
解决办法:升级pip 或者 使用 镜像源
问题:依旧下载超时
解决办法:官网找到下载包https://download.pytorch.org/whl/torch_stable.html
使用迅雷下载到本地
之后在终端打开下载到本地的路径,然后安装whl文件即可安装成功
安装成功后,可以直接
pip install torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
即可安装成功
尝试验证torch和cuda安装好没有:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
x = torch.randn(1)
if torch.cuda.is_available():
device = torch.device("cuda")
y = torch.ones_like(x, device=device)
x = x.to(device)
z = x + y
print(z)
print(z.to("cpu", torch.double))
结果:
参考博客:
https://blog.csdn.net/qq_35896136/article/details/106205990