ubuntu下pytorch环境搭建(后续再更新)

目前没太多时间,后续再更新,先把搭建顺序记录下来,不太确定有些版本号,如果有误,请告知,因为是上个月装的,现在按印象记录下来方便以后搭建。
**

ubuntu下pytorch环境搭建

**
第一步:安装对应nvidia显卡驱动,这里一定要保证两点,才算安装成功,第一是在终端输入nvidia-smi,可以显示对应显卡信息,第二是在设置-详情中可以看到显卡的型号,而不是cpu自带的集成显卡型号

第二步:安装nvidia显卡对应的cuda,具体的版本还需要根据后续安装的cudnn来决定cuda的版本,所以需要同时看好cudnn和cuda的版本对应关系,nvidia显卡对应的cuda版本会比较高,cudnn可能会不支持,所以以cudnn版本为准安装cuda比较妥当。比如我的显卡驱动是436,对应cuda版本为10.2,但cudnn最高只支持10.1,所以安装10.1即可,显卡也是可以匹配这个版本cuda的

第三步,安装cudnn,与上一步一起,这里好像还涉及到gcc和g++的安装,具体干嘛用的有点记不清了,如果没有安装,后续还是需要的,先记录一下

第四步:anaconda安装,这步挺简单,官网下载安装即可

第五步:pytorch安装,这个在官网上没法下载,速度出奇的慢,可使用清华源镜像下载,具体步骤百度即可

第六步:安装IDE,pycharm或者jupyter,jupyter在anaconda安装好后就已经可以用了,pycharm也很好安装,官网下载安装即可

以上是安装顺序,具体步骤后续在更新

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值