cuda+cudnn+torch-GPU安装总结

下载地址

0、查看驱动对应的cuda版本 地址

1、cuda下载地址(需要查看英伟达的驱动)

2、cudnn下载地址(需要和cuda的版本对应起来)

3、torch-whl离线下载地址 (需要和cuda版本对应起来)


安装细节

一、cuda

1、查看本机显卡的驱动:桌面右键+英伟达控制面板+帮助+系统信息+驱动版本

2、检查与驱动配套的cuda版本

3、安装即可

二、cudnn

下载zip文件,解压,复制到cuda安装文件夹中

三、torch-whl

在 此东西存在的文件 目录 中输入cmd 回车

输入 pip install xxx.whl(此whl的名字,直接把文件拖进黑色界面中)

你可能感兴趣的:(安装,pytorch,pytorch,深度学习,python)