Ubuntu中pytorch安装过程+安装之前的环境配置

 

  • 预安装环境

确定你的系统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一直安装不成功,说环境不兼容,重装太麻烦了,现在的我还没有装好

之后继续。。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值