流程
1. 安裝 NVIDIA 顯卡驅動與相關套件
https://www.tensorflow.org/install/gpu#ubuntu_1604_cuda_10
2. 重開機
3. 測試
nvidia-smi
nvcc -V
4. 編譯測試工具,並進行測試
cd /usr/local/cuda-10.0/samples
sudo make
cd bin/x86_64/linux/release
sudo ./deviceQuery
sudo ./bandwidthTest
5. 安裝 tensorflow-gpu
sudo pip3 install tensorflow-gpu
6. 測試 tensorflow 能否抓到 GPU
python3
from tensorflow.python.client import device_lib
def get_available_gpus():
local_device_protos = device_lib.list_local_devices()
return [x.name for x in local_device_protos if x.device_type == 'GPU']
get_available_gpus()
7. 若想防止未來誤更新,可使用 apt-mark 將套件 hold 住
sudo apt-mark hold nvidia-418 cuda-10-0 libcudnn7 libcudnn7-dev