anaconda创建虚拟环境安装pytorch
注意:
- 在安装之前请选择存储空间大于20G
- 尽量不要移动安装好的anaconda,一些快捷方式会找不到,需要重新配置环境变量
- 安装完成后我觉得虚拟机是个好东西,因为我一直在担心删除anaconda后再安装会不会安装失败等问题,随时准备重装系统
正文:
-
安装anaconda,使用权限选all users,不自动添加环境变量,随后手动添加
-
win+r命令行,或者Anaconda Prompt
-
查看镜像源 conda config --show channels #注意两个符号
-
(如需要替换)删除所有镜像源 conda config --remove-key channels
-
添加镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
-
查看当前已有虚拟环境 conda env list
-
创建虚拟环境 conda create -n env_name python=3.8 #自己python版本,直接在命令行输入python会显示版本
-
进入虚拟环境 conda activate env_name
-
安装pytorch,打开pytorch官网,向下拖动,找到下载的命令部分,选择符合电脑版本的选项,将命令复制到命令行执行,可以将命令中 -c删掉,否则是从默认源下载,速度慢。下载选择conda和pip应该都可以,本人选择conda,CUDA只能选择比自己电脑版本低的下载,在命令行输入 nvidia-smi 显示电脑显卡信息,如果没有nvidia 可以选none或者cpu,版本不够有两种选择:1.升级因特尔显卡驱动,比如本人CUDA升级后由11.1变成11.6;2.在pytorch官网下载原来的版本 (这里下载很慢,文件比较大)
-
安装完成,检验是否可以使用CUDA加速,返回true,成功
python import torch torch.cuda.is_available()