在ubuntu20.04下搭建深度学习环境(pytorch1.5)

  首先声明,完成这个环境的搭建不是我一个人的努力,是我在网上查找好多相关资料并得益于他们的帮助完成的,在下文对应的地方会放上相关链接。整个过程我将它分为4个步骤。

一、在ubuntu20.04上安装python3.7;

  不知道为什么,我下载的这个ubuntu20.04没有自带python环境,刚开始我也惊了一下,不过不管那么多了,装一下也还好啦。故自行安装python3.7,简单粗暴:sudo apt-get install python。它会自动安装最新版python。

二、检查硬件并安装对应的显卡驱动;

  参考链接如下:https://blog.csdn.net/qiancaobaicheng/article/details/95096354

  首先使用ubuntu-drivers devices命令查看自己的机器所需要安装的驱动版本,比如我这个TITAN RTX(型号)就需要安装430或者440。然后参照上面链接中的信息,在官网上下载对应的驱动并安装即可。

   注:目前最新的ubuntu20.04系统在选择驱动的时候,选用ubuntu18.04或者19.04的都可以。

三、安装cuda、cudnn并检查是否正确安装;

     关于cuda、cudnn的安装,这篇博文写得很好,大家可以参考安装https://blog.csdn.net/ashome123/article/details/105822040/。我在这里就不再赘述了。但需要注意的是中间的gcc降级,一般可能会想不到这个,因为在windows下应该是不需要这个操作的(据我所知)。

  成果图:

四、利用清华镜像安装pytorch1.5。

  为什么要用清华镜像安装pytorch1.5呢?因为官网直接在线安装提示timeout(连不上);又没有whl文件供下载(晚点应该会有)。所以只能考虑用别的镜像源了,最终在网上找到了这个。

  conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  conda config --set show_channel_urls yes

  conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

  conda install pytorch torchvision cudatoolkit=10.0

  原谅链接:https://blog.csdn.net/zzq060143/article/details/88042075

  因为我安装的是cuda10.2,所以我将上述最后一行的参数“cudatoolkit=10.0”修改为“cudatoolkit=10.2”就可以了,也是支持GPU的。不知道评论区里说的不支持GPU是什么情况,感谢这位博主。

  截图为证:

   终于站在巨人的肩膀上,完成了ubuntu20.04环境下深度学习环境python3.7.7、cuda、cudnn、pytorch1.5的安装,在此对以上引用到博文的博主一并表示感谢!

  最后,欢迎大家提出疑问,批评指正!

 

posted on 2020-06-17 15:57  jinjunweina  阅读(4875)  评论(0编辑  收藏  举报

导航