配置:win 10 ,python=3.6
安裝pytorch-1.1.0,cudatoolkit-9.0,torchvision-0.3.0.
出現的問題:import torch 的時候,出現了OSError ,找不到指定模塊。於是在網上試了很多方法,包括安裝dll文件,都沒用。后來卸載子環境重裝,我懷疑是先安裝了openCV,再安裝pytorch出現了沖突。於是重新找了篇帖子,安裝成功。安裝步驟如下。
1、打開Anaconda Prompt,輸入命令:
conda create -n shuzishibie python=3.6(這里是子環境的名字,可任取)
2、然后激活子環境,輸入:
conda activate shuzishibie
3、進入子環境后,輸入:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch,這里安裝報錯,因為官方給出的安裝命令中最后的-c pytorch,是默認指定anaconda官網上的安裝源,可能由於牆的問題,無法正常或者迅速下載。所以我這里選用清華源。
4、直接在命令行中指定,將anconda的安裝源修改為清華的鏡像源。輸入:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda
config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda
config --set show_channel_urls yes
5、然后執行pytorch的安裝命令,輸入:
conda install pytorch torchvision cudatoolkit=9.0
6、最后,會發現已安裝成功。測試一下。輸入:python;測試一下:import torch
7、完美運行!