pip install torch报错

问题：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