windows下安装PyTorch，利用conda或者p

windows下安装PyTorch

使用anaconda的虚拟环境来安装，前提是安装好了anaconda。

利用conda或者pip 安装PyTorch

1.首先创建一个虚拟环境，这里我创建的虚拟环境命名为dgnn

conda create -n dgnn python=3.8

-n后面的参数指定虚拟环境名字，python后可以指定版本

通常创建下载速度会比较慢，可以添加镜像来加速

conda create -n dgnn python=3.8 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

-c后面的参数指定镜像地址，常见的镜像地址如下

删除虚拟环境

删除虚拟环境下的所有东西，这里删除掉虚拟环境torch以及其中所有的东西

conda remove -n torch --all

确定显卡型号——GeForce RTX 3050

确定显卡算力——8.6

https://en.wikipedia.org/wiki/CUDA

查看自己的CUDA Driver Version——12.0

nvidia-smi

确定CUDA Runtime Version——11.6

打开PyTorch官网，使用cuda命令安装

conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.6 -c pytorch -c nvidia

pip命令安装

pip install torch==1.13.0+cu116 torchvision==0.14.0+cu116 torchaudio==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu116

验证PyTorch

返回true就证明PyTorch安装成功

import torch
print(torch.cuda.is_available())