主要步驟:
0.安裝anaconda3(基本沒問題)
1.配置清華的源(基本沒問題)
2.查看python版本,運行 python3 -V; 查看CUDA版本,運行 nvcc -V
3.如果想用最新版本的python,可以創建新的python版本:
conda create --name python38 python=3.8
conda activate python38
此時,運行python3 -V,會顯示python的版本為3.8。再運行
conda deactivate
python的版本會恢復為原來的,如3.7
4.根據官網(https://pytorch.org/),獲取命令行。此時要注意,如果要用清華的源,需要命令行中的“-c pytorch”;然后將python版本添加到命令行中,如“python=3.8”
在我本地,運行的命令是“conda install pytorch torchvision torchaudio cudatoolkit=10.2 python=3.8”
在我這安裝時,一開始是使用python3.7,但安裝完之后,運行 import torch 會有 ModuleNotFoundError: No module named 'typing_extensions' 的錯誤,然后發現安裝pytorch前后,python的版本由3.7.0變為3.7.9了。最后試着創建python 3.8,安裝pytorch時指定python版本為3.8,嘗試import torch后沒問題。