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

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

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

conda create --name test python=3.8 -y

test为环境名

2.激活虚拟环境

conda activate test

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

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 --add channels https://mirrors.aliyun.com/anaconda/pkgs/free

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

Conda配置阿里源

conda config --set show_channel_urls yes

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

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

conda install pytorch torchvision torchaudio cudatoolkit=11.3

我使用pytorch提供的语法

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

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

python

import torch

torch.cuda.is_available()

显示如下即为成功

附
删除conda虚拟环境方法

conda remove -n test --all

重命名虚拟环境

conda create -n newtest --clone test