zoukankan      html  css  js  c++  java
  • 在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的安装,在此对以上引用到博文的博主一并表示感谢!

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

  • 相关阅读:
    spring 自定义事件发布及监听(简单实例)
    解析spring中的BeanFactory(看完会有收获)
    如何提高锁的性能
    spring MVC模式拦截所有入口方法的入参出参打印
    java基于feemarker 生成word文档(超级简单)
    数据库事务特性汇总
    如何让window.open()以post请求方式调用(巧妙解法)
    a标签添加背景图片的解决办法
    深入理解Django Admin的list_display, list_filter和raw_id_fields,filter_horizontal选项
    django配置log日志
  • 原文地址:https://www.cnblogs.com/jinjunweina/p/13139419.html
Copyright © 2011-2022 走看看