conda pytorch 配置

主要步骤:

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后没问题。

 

posted @ 2020-11-13 00:04  我是包子  阅读(2017)  评论(0编辑  收藏  举报