Ubuntu下安装PyTorch杂记

最近几天我一直常用的Kubuntu(KDE yes!)更新至22.04后居然出现无法更改软件源的bug,去Kubuntu论坛一看有同样问题的人还不在少数,但却没有好的解决办法,故而只有备份数据装回Ubuntu。

由于学习需要,我需要再装一遍Pytorch。Pytorch在linux上安装推荐使用conda,使用conda安装会打包cudatoolkit等一系列必要的软件包,不过有一点值得注意,一定要换源,清华上交的都可以,而且要删除config文件里的–default,不然还是会使用默认源,我个人的计算机会在进度50多的时候显示网络问题终止安装。不过安装完成后仍有问题,第一是:

显示没有安装nvidia-tool-kit,第二是因为ubuntu22.04是前天才发布的版本,nvidia驱动没有自带。总而言之

返回的是False。

对于第二个问题,它很容易解决。去NVIDIA官网搜索你的显卡对应的驱动安装,后来我比对了一下,在终端里打出来:

[En]

For the second problem, it is easy to solve. Go to Nvidia’s official website to search for the corresponding driver installation of your video card, and later I compared it and typed it in the terminal:

是可以找到官网提供的显卡驱动,我选择的是版本号510的最新稳定驱动。使用 apt-get install指令安装即可

然后使用:

可以看到自己显卡的CUDA版本,我的是11.6,虽然我在conda里确实看到了cuda等一系列包已经正确配置。没办法,我只得再去官网重新下载cudatoolkit对应的版本,照着上面的指示安装即可,安装完成后

返回值为True,这个过程大概耗了我一个下午。

Original: https://www.cnblogs.com/BensonLau/p/16184095.html
Author: BensonLau
Title: Ubuntu下安装PyTorch杂记

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/499784/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球