主要步骤:
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后没问题。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:conda pytorch 配置 - Python技术站