anaconda安装pytorch
添加清华镜像
-添加镜像
清华镜像https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
-显示检索路径
conda config --set show_channel_urls yes
-检查一下看是否添加上了镜像
进入anaconda-environments-channels,应该会有刚才add的镜像,并删除default
创建自己的pytorch环境
-打开anaconda,进入environments
-点击create,出现如下图所示
Name由自己定义(此处我命名为Pytorch),Python版本根据自己下载版本设定(会自动选择)
-点击create,等待创建环境
完成后,会有base和Pytorch两个环境
Pytorch安装
-根据自己的安装版本,在pytorch网站寻找代码
进入NVIDIA控制面板,查看CUDA驱动版本(我的是11.1.96)
进入官网:https://pytorch.org/
复制代码:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
将复制的代码粘贴到命令行格式下(我修改了版本号11.1),弹出提示,输入y,即可完成安装,显示“done”
测试
-激活pytorch环境: conda activate Pytorch(如果已经在这个环境中,既可以省略此步骤)
-进入python: python
-测试pytorch: import torch
import tochvision
显示为上图,没有报错,就表示安装成功!