在Ubuntu上配置和測試GPU加速需要進行以下步驟:
sudo apt update
sudo apt install nvidia-driver
sudo apt install nvidia-cuda-toolkit
安裝cuDNN:cuDNN是NVIDIA提供的用于深度學習框架的加速庫。可以通過NVIDIA的官方網站下載cuDNN,并按照官方文檔進行安裝。
配置環境變量:在.bashrc文件中添加CUDA和cuDNN的路徑到環境變量中。可以通過以下命令編輯.bashrc文件:
nano ~/.bashrc
然后在文件末尾添加以下內容:
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH
保存并退出文件,然后運行以下命令使配置生效:
source ~/.bashrc
nvidia-smi
如果輸出了GPU的信息,則說明GPU加速已經配置成功。
通過以上步驟,你可以在Ubuntu上配置和測試GPU加速。希望對你有幫助!