最近幾天我一直常用的Kubuntu(KDE yes!)更新至22.04後居然出現無法更改軟體源的bug,去Kubuntu論壇一看有同樣問題的人還不在少數,但卻沒有好的解決辦法,故而只有備份數據裝回Ubuntu。 由於學習需要,我需要再裝一遍Pytorch。Pytorch在linux上安裝推薦使用co ...
最近幾天我一直常用的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,這個過程大概耗了我一個下午。