Pytorch安装时的坑！

本文链接：https://www.cnblogs.com/chenzhihong294/p/14716453.html

---

1、pytorch安装好后依然报错：no moduled name 'torch'

网上很多教程是在anaconda prompt当中安装的pytorch，所用的命令行创建了一个名为pytorch_gpu的虚拟环境:

conda create -n pytorch_gpu pip python=3.8

所以命令行中的安装信息，具体的安装位置是：

environment location:D:/Anaconda/envs/pytorch_gpu

所以安装的pytorch_gpu是放在D:/Anaconda/envs/pytorch_gpu这个文件夹下面的，但是python能够导入的包全都放文件夹D:AnacondaLibsite-packages中

解决方法：将D:/Anaconda/envs/pytorch_gpu/Lib/site-packages下面的四个文件夹复制到D:AnacondaLibsite-packages中：

2.报错：OSError: [WinError 126] 找不到指定的模块caffe2_detectron_ops_gpu.dll” or one of its dependencies

这个错误提示的是找不到caffe2_detectron_ops_gpu.dll，根据路径找到我们本地宝中这个文件尾缀前面加了gpu。

该错误的原因是未安装cudatoolkit（网上大部分教程均为手动安装cuda和cudnn）。

解决方法：把缺少的cudatoolkit和其他包重新安装。

conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch

若用本地镜像源，删除后面的 '-c pytorch'即可，至此便可解决该问题

---

测试代码：

import torch
flag = torch.cuda.is_available()
print(flag)

ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

预期结果：

D:Anacondapython.exe F:/Py_Projects/DeepLearning/Pytorch.py
True
cuda:0
GeForce GTX 1050 Ti
tensor([[0.8742, 0.0954, 0.2961],
        [0.4184, 0.2107, 0.9721],
        [0.5613, 0.5419, 0.9607]], device='cuda:0')

Process finished with exit code 0

成功！