Ubuntu下安装PyTorch杂记

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

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

nvcc -V

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

torch.cuda.is_available（）

返回的是False。

对于第二个问题，很容易解决，去英伟达官网搜索自己显卡对应的驱动安装即可，而且后来我对比了一下，在终端中输入：

ubuntu-drivers devices

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

然后使用：

nvidia-smi

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

torch.cuda.is_available（）

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