Win11+RTX3060配置pytorch环境(一些踩坑记录)

    30系显卡目前只支持cuda11以上的版本，如果装的版本过低会导致Pytorch无法使用GPU加速。装错了就得卸载重装一遍。
目前我装成功的环境版本是：

• CUDA Toolkit 11.1.1
• ​cuDNN v8.1.1 (Feburary 26th, 2021), for CUDA 11.0,11.1 and 11.2
• pytorch版本1.8.0

去pytorch官网可以找到对应版本的pytorch版本1.8.0
https://pytorch.org/get-started/previous-versions/
但是在这里推荐用pip安装，因为conda安装巨慢，如果你用清华镜像，他那边目前只有cpu版本的pytorch，装上你会发现用不了cuda加速（巨坑）。

pip安装相对速度快很多

# CUDA 11.1
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

最后测试：

python
import torch
torch.cuda.is_available()

返回True表示安装成功最后

如果觉得帮到你了，点个赞再走吧