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

posted @ 2024-08-03 13:51  Augustone  阅读(117)  评论(0)    收藏  举报