Ubuntu18.04 顯卡驅動+Cuda安裝踩坑記錄 以及Ubuntu虛擬內存的添加

前幾天買了張亮機卡，終於把主顯卡成功直連到Unraid OS的虛擬機上了。然后就開始安裝ubuntu系統開始配置環境，遇到了不少坑，特此記錄。

1. gcc版本問題

   在安裝顯卡驅動的時候，不要修改gcc版本，使用與內核對應的版本（默認版本，1804應該是7.3.0）。否則安裝驅動會出現編譯失敗的問題。

2. 新版本驅動自帶Cuda10.1

   裝完驅動以后，使用nvidia-smi會發現看到Cuda10.1，不要慌，按照正常流程再裝一邊Cuda即可，然后再配置環境變量。

3. 重裝顯卡驅動

   如果驅動安裝有問題，卸載重裝。18.04和16.04重啟圖形界面的指令不同，不要用錯了。我是直接裝的NVIDIA網站上下載的驅動。自動安裝確實會方便很多，但是萬一出錯了就不知道是哪里有問題了，很麻煩。

 

接下來就是在跑訓練的時候遇到的一個小問題。數據量特別大的情況下，內存不夠用，會出現Memory Error的問題。排除所有其他問題，最終發現內存不足。Ubuntu在安裝的時候默認給我分配了1GB的虛擬內存，太小，於是增加一個swap文件。

sudo fallocate -l xxxG /swapfile   #<xxx>用多少就寫多少
sudo chmod 600 /swapfile
sudo mkswap /swapfile
sudo swapon /swapfile

如果發現內存仍然不夠，可以擴容。　　

sudo swapoff /swapfile
sudo dd if=/dev/zero of=/swapfile bs=1024 count=xxxG   #<xxx>用多少就寫多少
sudo swapon /swapfile

如果擴容大的話，會占用比較長的時間，取決於硬盤速度。