这个教程在b站和网上很多,但是随着软件更新,包的冲突问题也越来越明显,所以在这里记录我的版本

1.查看自己的GPU

在anaconda powershell Prompt中输入

nvidia-smi.exe

结果:

我的cuda版本是12.5

我的GPU是RTX3070

2.看python的版本号

3.下载pytorch:

去pytorch的官网,根据cuda下载pytorch

验证即可

参考:

Start Locally | PyTorch

Windows 下安装 CUDA 和 Pytorch 跑深度学习 - 动手学深度学习v2_哔哩哔哩_bilibili

[env]pytorch_lightning和pytorch对应版本问题_pytorch-lightning版本对应-CSDN博客

更多推荐