Windows安装cuda个cudnn
注:
在装cuda前其他nvidia软件可以都卸载了,cuda会装的很全面。
在安装cuda之前实际上需要安装visual studio社区版即可。这是cuda安装的前提。
1,下载两个文件,cuda和cudnn。注意cuda与cuddn之间有兼容匹配关系。
2,安装cuda,最好自定义安装到其他盘(但附带安装的软件仍被安装在c盘,英伟达真tm傲慢啊)
3,安装好cuda后解压cudnn,既然是解压,真是够垃圾的方式。解压后还要手动拷贝到cuda相应的目录下。
4,最后用extras下两个可执行文件测试一下,能找到gpu。
5,安装pytorch
a,卸载pytorch,因为它可能是cpu版本的。
import torch
print(torch.__version__)
若为cpu版本的则,pip uninstall torch
b,重新安装,注意,命令在这里:PyTorch
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
浙公网安备 33010602011771号