最近幾天我一直常用的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，這個程序大概耗了我一個下午，
轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/462912.html
標籤：其他
上一篇：python命名空間和作用域