Anaconda + pytorch+GPU+cuda11.3

Anaconda中创建新的虚拟环境
1.基本指令

①检验是否安装及当前conda的版本：conda -V
②查看已安装的包conda list
③查看当前存在哪些虚拟环境conda env list
④删除虚拟环境conda env remove --name your_env_name

2.Python创建虚拟环境

第一步：创建python3.6的环境，虚拟环境命名为pytorch_gpu，

conda create -n pytorch_gpu python=3.6

可以选择Ctrl+c可以 停止当前操作
等待创建虚拟环境

选择Y之后，虚拟环境就下载好了。检查当前已安装的虚拟环境conda env list，此时虚拟环境确实存在。

第二步 安装GPU版本pytorch的CUDA和cudnn版本
打开命令行终端WIN+R 输入cmd

输入命令nvidia-smi 检查已下载的CUDA版本 本机为CUDA版本11.6`

检查本机下载的CUDA版本(以这个为准）

输入命令nvcc --version 检查已下载的CUDA版本 本机为11.3

检查cudnn版本 8.6.0.163

根据CUDA和cudnn版本信息
进入虚拟环境,根据CUDA和cudnn版本信息下载对应的gpu版本pytorch(下载需要一定时间）
退出当前虚拟环境conda deactivate

activate pytorch_gpu
pip install torch==1.9.0+cu113 torchvision==0.10.0+cu113 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

下载完成后conda list查看虚拟环境资源包

之后输入

python
import torch
torch.cuda.is_available()

结果为True表明GPU版本的pytorch虚拟环境已经搭建完成。