Linux操作系統安裝CUDA工具包
更新時間 2023-12-28 17:26:40
最近更新時間: 2023-12-28 17:26:40
分享文章
本節主要介紹了Linux操作系統安裝CUDA工具包的流程。
以下操作以Ubuntu 16.04 64bit操作系統,GPU實例使用run包安裝CUDA 10.1為例。
- 登錄云主機。
- 在CUDA下載頁面中,按照Tesla驅動及CUDA工具包獲取方式中的對應的索引項在頁面中進行選擇。
圖 選擇CUDA的版本

- 選擇完成后,頁面會自動呈現出Ubuntu 16.04 64bit對應的CUDA 10.1的下載地址,復制下載地址。
圖 復制CUDA下載地址

- 在云主機內部執行如下命令進行下載。
wget 復制的鏈接地址
例如:wget //developer.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.105_418.39_linux.run
圖 Linux云主機下載CUDA

- CUDA工具包下載完成后,按照NVIDIA官網安裝指引進行安裝。
按照如下圖的NVIDIA官網的Installation Instructions進行安裝。
圖 Linux云主機安裝CUDA

- 執行以下命令安裝CUDA。
sh cuda_10.1.243_418.87.00_linux.run
- 在安裝界面輸入“accept”,回車進入下一步。
圖 安裝CUDA_1

- 選擇“Install”,回車進入安裝過程。
圖 安裝CUDA_2

圖 安裝CUDA完成

- 執行以下命令進入/usr/local/cuda-10.1/samples/1_Utilities/deviceQuery 目錄下。
cd /usr/local/cuda-10.1/samples/1_Utilities/deviceQuery
- 執行make命令,自動編譯deviceQuery程序。
- 執行以下命令查看CUDA是否安裝成功。
./deviceQuery
如回顯信息中有CUDA的版本信息證明CUDA安裝成功。
圖 deviceQuery回顯信息

- 執行以下命令查看CUDA的版本。
/usr/local/cuda/bin/nvcc -V
圖 查看CUDA的版本

- 執行以下命令啟動持久模式。
sudo nvidia-smi -pm 1
啟用持久模式設置可以優化Linux實例上GPU設備的性能。