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

本文详细介绍了如何在Ubuntu20.04上从零开始搭建深度学习环境,包括安装Python3.7,检查硬件并安装显卡驱动,安装CUDA和cuDNN,以及利用清华镜像安装PyTorch1.5,解决官方安装timeout的问题。最后,作者对提供帮助的博主表示感谢。
摘要由CSDN通过智能技术生成

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

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

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

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

首先使用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

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

截图为证:

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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值