目前没太多时间,后续再更新,先把搭建顺序记录下来,不太确定有些版本号,如果有误,请告知,因为是上个月装的,现在按印象记录下来方便以后搭建。
**
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也很好安装,官网下载安装即可
以上是安装顺序,具体步骤后续在更新