Windows系统安装PyTorch：CUDA版+CPU版

Windows系统安装PyTorch

PyTorch官网点击这里：PyTorch官网

创建虚拟环境

打开Anaconda Prompt，创建虚拟环境，如创建一个名字为pytorch的运行环境
conda create -n pytorch python=3.9
查看环境
conda info --envs
切换环境
conda activate pytorch
删除环境
conda remove -n pytorch --all

CUDA版（需要NVIDIA驱动程序）

检查CUDA Driver版本

打开Anaconda Prompt，输入
nvidia-smi

加载已有whl安装包

pip install "torch-1.7.1+cu101-cp39-cp39-win_amd64.whl" -i [https://pypi.tuna.tsinghua.edu.cn/simple](https://pypi.tuna.tsinghua.edu.cn/simple)
pip install "torchvision-0.8.2+cu101-cp39-cp39-win_amd64.whl" -i [https://pypi.tuna.tsinghua.edu.cn/simple](https://pypi.tuna.tsinghua.edu.cn/simple)

官网下载

如果官网没有对应的CUDA版本，可以找以前的版本下载

例如，查找CUDA11.7版本

~~ 尝试过官网下载，有点慢！ ~~

CPU版（适用于非英伟达显卡）

pip3 install torch torchvision torchaudio

检验安装

输入python
输入import torch
输入print(torch.cuda.is_available())
得到True则安装成功

非英伟达显卡得到False表示安装成功，没有显卡会在CPU上计算数据