level 2
1. 查看当前nvidia驱动的版本号 以及支持的cuda版本
cmd : 输入 nvidia-smi
2. 下载对应的cuda toolkit 11.1
3. 下载对应的cudnn ,解压复制
4. download.pytorch.org/whl/torch_stable.html ,到此网页中找到对应的cuda版本的安装包,可以下载,主要是为了查看相对应的python版本
比如: cu121/torch-2.2.0%2Bcu121-cp312-cp312-win_amd64.whl ,这个包会很大,3G多,可下载再离线安装
cu121 : cuda12.1
cp312: python3.12
5. 在线安装
建立对应版本的python虚拟环境,然后运行pip就可以
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu121
6. 总结: python + cuda 这两个的版本需要对应,其他步骤都很容易
2026年02月17日 10点02分
