Ubuntu 20.04+Nvidia 驱动365+cuda 11.3安装PYtorch

辛辛苦苦装系统,搞了两天终于成功。记录一下安装过程。大体上的下面的教程还是对的。但是还是有一些坑在里面。

参考教程如下:Ubuntu 20.04安装CUDA 11.0、cuDNN 8.0.5、PyTorch 1.7.0 - BooTurbo - 博客园

但是需要注意以下地方,先安装显卡驱动,在安装cuda的时候,有一项选择是否安装显卡驱动,如下图:

 第一个CUDA Driver 一定把X去掉,不然两个显卡驱动会冲突,导致灰屏,我就是这个原因重新装了系统,才定位出原因。

第二个cuda 版本和pytorch版本一定要对应,我先安装了cuda11.5后来发现pytorch 支持11.3不支持11.5所以重新安装显卡驱动和cuda,浪费很多时间。

第三个需要注意,用conda 安装Pytoch的时候一定要先建立虚拟环境,命令如下:

conda creat -n Pytorch

conda activate Pytorch

在虚拟环境中执行如下命令才能安装成功:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

当然也可以切换清华源。

希望踩过得坑对别人有帮助。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值