- 预安装环境
确定你的系统driver version
具体方法: 输入语句 nvidia-smi 得到如下结果
获得driver version 与cuda关系: https://stackoverflow.com/questions/30820513/what-is-the-correct-version-of-cuda-for-my-nvidia-driver/30820690#30820690
得到CUDA cudNN的对应关系
cuda、cudNN如何下载安装: http://gwang-cv.github.io/2018/06/28/Ubuntu下查看CUDA版本、cudnn版本号/
下载对应的cuda官网:https://developer.nvidia.com/cuda-downloads
cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-download,需要注册账号
在安装之前,先检查一下是否有cuda,通常cuda不用装,cudann是自己装
- pytorch安装
在pytorch官网: https://pytorch.org
Ubuntu下查看CUDA版本、cudnn版本号: http://gwang-cv.github.io/2018/06/28/Ubuntu下查看CUDA版本、cudnn版本号/
根据得到的版本信息,选择对应的安装语句:
参考:
确定你的系统driver version: https://blog.csdn.net/qq_15999617/article/details/83184221
获得driver version 与cuda关系: https://stackoverflow.com/questions/30820513/what-is-the-correct-version-of-cuda-for-my-nvidia-driver/30820690#30820690
得到CUDA cudNN的对应关系: https://blog.csdn.net/yuejisuo1948/article/details/81043962
Ubuntu下查看CUDA版本、cudnn版本号: http://gwang-cv.github.io/2018/06/28/Ubuntu下查看CUDA版本、cudnn版本号/
Ubuntu 非管理员权限 卸载安装cuda和cudnn版本: https://blog.csdn.net/Miya_coding/article/details/82666790
一定要主要cuda、cudNN之间的版本对应啊,不然就是现在的我,pytorch一直安装不成功,说环境不兼容,重装太麻烦了,现在的我还没有装好
之后继续。。。。