前言

我目前已有的配置：

Windows10 + Anaconda3 + Python 3.8 + CUDA10.2 + cuDNN

最近在跑一個(gè)人眼視線的模型，需要import torch， torchvision. 我之前已經(jīng)在pytorch環(huán)境中安裝了pytorch 和torchvision了，打開(kāi)Anaconda prompt, 在python中可以import torch 和 torchvision，但是在pycharm中始終無(wú)法import. 后面尋找原因

發(fā)現(xiàn)了兩個(gè)問(wèn)題：

• 1.我的CUDA, cudnn, pytorch , torchvision, python安裝的版本都不匹配
• 2.在Pycharm中import torch報(bào)錯(cuò)，原因項(xiàng)目的解釋器路徑是Python的安裝路徑，需要改為Anaconda的Python的路徑即可，而且我之前的pytorch， torchvision并不是安裝在我的anaconda中的python環(huán)境中。

第二個(gè)原因應(yīng)該是pycharm無(wú)法正常import torch 和torchvision的主要原因。

為了之后的工作方便，后面我還是將這些工具全都重新安裝了一遍，最后的版本為 CUAD 10.2 ---- pytorch 1.6.0 ----torchvision 0.7.0。

如果你也遇到了這個(gè)pycharm 的import的問(wèn)題的話，建議你先處理第二個(gè)原因試一下

下面我記錄一下我的解決辦法：

一、anaconda下載pytorch, torchvision包

打開(kāi)cmd窗口，用CD命令，將當(dāng)前目錄定位到anaconda3目錄下（我安裝的是anaconda3），再用dir命令查看anaconda3中的文件夾，你會(huì)發(fā)現(xiàn)anaconda3文件夾下面有Scripts這個(gè)文件夾的

用cd命令，切換到Scripts文件夾中，然后我們就可以使用pip來(lái)進(jìn)行安裝了（pip是ananconda的pip工具）

輸入命令 pip install torch 和pip install torchvision， 然后等待安裝完成即可。

因?yàn)槲业哪Ｐ椭泻罄m(xù)還需要import cv2, 所以我順便把cv2的包也安裝完了

輸入pip install opencv-python, 完成了opencv的安裝。

二、Pycharm中import torch, import torchvision報(bào)錯(cuò)

在Pycharm中import torch報(bào)錯(cuò)，原因項(xiàng)目的解釋器路徑是Python的安裝路徑，改為Anaconda的Python的路徑即可

具體步驟為：

File–Settings–Project:xxx–Project Interpreter–show all–點(diǎn)擊“+”添加–System Interpreter–C:\Users\Administrator\Anaconda3\python.exe

問(wèn)題得到解決。

總結(jié)

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評(píng)論