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:\Anaconda\Lib\site-packages中

解決方法：將D:/Anaconda/envs/pytorch_gpu/Lib/site-packages下面的四個文件夾復制到D:\Anaconda\Lib\site-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:\Anaconda\python.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

成功！