conda/miniconda 安装pytorch(GPU)_Windows_实录_2022.1.4_miniconda 阿里源

conda/miniconda 安装pytorch(GPU)_Windows_实录_2022.1.4

1.打开anaconda promopt，创建虚拟环境

conda create --name test python=3.8 -y
1

test为环境名

2.激活虚拟环境

conda activate test
1

进入test环境
3.添加镜像源（安装torch时 不推荐不推荐不推荐）直接转4
清华镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
1

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
1

阿里云

conda config --add channels https://mirrors.aliyun.com/anaconda/pkgs/free
1

conda config --add channels https://mirrors.aliyun.com/anaconda/pkgs/main
1

Conda配置阿里源

conda config --set show_channel_urls yes
1

4.安装torch
打开pytorch官网，根据自己nvidia驱动以及cuda版本选择合适的pytorch版本

注意：使用国内镜像源需要删除尾部“-c pytorch”

conda install pytorch torchvision torchaudio cudatoolkit=11.3
1

我使用pytorch提供的语法

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
1

5.安装后验证：
先进入python编译器环境

python
1

import torch
1

torch.cuda.is_available()
1

显示如下即为成功

附
删除conda虚拟环境方法

conda remove -n test --all
1

重命名虚拟环境

conda create -n newtest --clone test
1