1.下載Anaconda3-5.3.1-Linux-x86_64.sh並運行如下命令,按提示操作完成后退出、重進命令行。
bash Anaconda3-5.3.1-Linux-x86_64.sh
2.創建新環境,命令中的yourEnv指新創建的虛擬環境的名字,可自己命名。
conda create --name yourEnv python=3.7
3.進入(或激活)所創建的虛擬環境。備注:除自己創建外,默認或自帶的虛擬環境叫做base。
source activate yourEnv
4.進入PyTorch獲取命令,然后執行。
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch//根據需要選擇對應的命令。
5.在4中運行的命令可能會報錯,此時執行下面的命令,之后再執行4中的命令。
conda update -n base -c defaults conda
6.大功告成,注意:每次重新進入新的命令行窗口時,都要激活對應的虛擬環境,這樣才能使用里面的pytorch,激活成功之后,命令行開端會出現(yourEnv)的符號。