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的安装,在此对以上引用到博文的博主一并表示感谢!

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

  • 相关阅读:
    Redmine-java-api使用
    Junit4 基于 custom Rule retry
    Extentreports在安卓中的应用
    uiautomator 2.0 自定义testrunner使用
    uiautomator 启动原理
    uiautomator 自定义注解的应用
    uiautomator 原理 (UiAutomation、UiAutomatorBridge、QueryController)
    uiautomator 自定义testrunner使用和启动原理
    HeadFirstPython-文件与异常
    HeadFirstPython-初识python
  • 原文地址:https://www.cnblogs.com/jinjunweina/p/13139419.html
Copyright © 2011-2022 走看看