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

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

  • 相关阅读:
    Java中的位运算符
    华为2019年NE40E-X8,承诺命令
    华为网络设备修改console密码
    MySQL5.7 多实例
    华为防火墙域间策略全开命令
    Linux基础命令---ntpstat显示时间服务器同步
    【转载】Java程序模拟公安局人员管理系统
    redis cluster集群动态伸缩--删除主从节点
    DBUtils的使用之查询的操作
    服务治理-Resilience4j(限流)
  • 原文地址:https://www.cnblogs.com/jinjunweina/p/13139419.html
Copyright © 2020-2023  润新知