GPUが起動できるかのチェック

GPU 利用できるかのチェック

1. 以下を実行し、Trueが出ているかチェック

import torch

print(torch.cuda.is_available())

2. 以下Falseが出たとき用

システムのアップデート

sudo apt update
sudo apt upgrade
sudo apt dist-upgrade

CUDA, CUDNNの削除

sudo apt --purge remove nvidia-*
sudo apt --purge remove cuda-*

Install CUDA

今回は11.3のインストール例

wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run

再起動

reboot

Install できたかチェック

import torch
print(torch.cuda.is_available())

参考

How to install the NVIDIA driver