Cuda version 관리하기

Linux/Ubuntu

jinmc 2021. 4. 27. 13:41

Ubuntu 20.04를 사용하다가, Ubuntu 버전에 한계를 느껴서, 18.04로 다운그레이드 하였습니다.

20.04에서는 Deepstream도 돌아가지 않고, Tensorflow를 돌리는 데에도 여러가지 문제가 있었기 때문입니다.

하지만 18.04를 설치하고 나니 Cuda가 9.1로 설치가 된 것을 확인하였습니다.

Cuda 9.1이 호환되지 않는 텐서플로우가 있는 만큼, Cuda를 10.1로 업그레이드 하기로 하였습니다.

하지만 사실 여러 개의 Cuda 버전을 관리할 수도 있다고 합니다.

위 블로그에서는

$ sudo apt-get install cuda-toolkit-10-1

로 가능함을 볼 수 있습니다.

이미 cuda toolkit을 install 하였다면, 다음 커맨드로 지울 수 있습니다.

$ sudo apt-get remove nvidia-cuda-toolkit