anaconda创建虚拟环境安装pytorch


注意:

  1. 在安装之前请选择存储空间大于20G
  2. 尽量不要移动安装好的anaconda,一些快捷方式会找不到,需要重新配置环境变量
  3. 安装完成后我觉得虚拟机是个好东西,因为我一直在担心删除anaconda后再安装会不会安装失败等问题,随时准备重装系统

正文:

  1. 安装anaconda,使用权限选all users,不自动添加环境变量,随后手动添加

  2. win+r命令行,或者Anaconda Prompt

  3. 查看镜像源 conda config --show channels #注意两个符号

  4. (如需要替换)删除所有镜像源 conda config --remove-key channels

  5. 添加镜像源

    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/
    
  6. 查看当前已有虚拟环境 conda env list

  7. 创建虚拟环境 conda create -n env_name python=3.8 #自己python版本,直接在命令行输入python会显示版本

  8. 进入虚拟环境 conda activate env_name

  9. 安装pytorch,打开pytorch官网,向下拖动,找到下载的命令部分,选择符合电脑版本的选项,将命令复制到命令行执行,可以将命令中 -c删掉,否则是从默认源下载,速度慢。下载选择conda和pip应该都可以,本人选择conda,CUDA只能选择比自己电脑版本低的下载,在命令行输入 nvidia-smi 显示电脑显卡信息,如果没有nvidia 可以选none或者cpu,版本不够有两种选择:1.升级因特尔显卡驱动,比如本人CUDA升级后由11.1变成11.6;2.在pytorch官网下载原来的版本 (这里下载很慢,文件比较大)

    img

  10. 安装完成,检验是否可以使用CUDA加速,返回true,成功

    python         
    import torch 
    torch.cuda.is_available()