打开Anaconda Propmt
在命令行里面输入如下命令:
创建pytorch-gpu虚拟空间
conda create -n pytorch-gpu python=3.7 -y
切换到pytorch-gpu空间
conda activate pytorch-gpu
为了使得安装快速,需要添加清华源
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/conda-forge/
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/bioconda/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --set ssl_verify false
在如下页面(https://pytorch.org/)上获取安装命令:
安装
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
安装完以后,如下:代表安装成功