这个教程在b站和网上很多,但是随着软件更新,包的冲突问题也越来越明显,所以在这里记录我的版本
1.查看自己的GPU
在anaconda powershell Prompt中输入
nvidia-smi.exe
结果:
我的cuda版本是12.5
我的GPU是RTX3070
2.看python的版本号
3.下载pytorch:
去pytorch的官网,根据cuda下载pytorch
验证即可
参考:
Windows 下安装 CUDA 和 Pytorch 跑深度学习 - 动手学深度学习v2_哔哩哔哩_bilibili
[env]pytorch_lightning和pytorch对应版本问题_pytorch-lightning版本对应-CSDN博客