Windows11 Pytorch安装

背景介绍

大部分人都会用服务器来做深度学习，但那是到了一定体量的模型，学习过程中可以适当用自己电脑的显卡跑一下，那么windows虽然慢一点，但还是方便一点的。

基本环境

win11，anaconda，cuda11.8

Anaconda配置

channels:
  - defaults
show_channel_urls: true
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  deepmodeling: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/

由于更新过快难以同步，不同步pytorch-nightly, pytorch-nightly-cpu, ignite-nightly这三个包。所以不要用conda下载pytorch。

安装Pytorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118