nvidia 홈페이지에서 배포하는 cuda toolkit은 11.0부터 20.04를 지원하는데 tensorflow, torch에서는 아직 지원을 안한다.
(만약 20.04에서 18.04용 cuda toolkit을 사용할 경우 cuda 코드 컴파일시 문제가 생긴다고 한다.)
이런 경우 그냥 apt로 ubuntu 20.04 기본 레포지토리에 있는 cuda 10.1 패키지를 설치해서 사용하면 된다.
sudo apt install nvidia-cuda-toolkit
nvcc --version
위와 같이 설치 후 버전 확인을 하면 10.1이 출력되는것을 확인할 수 있다.
torch와 tensorflow는 아래와 같이 기존처럼 설치하면 된다.
pip3 install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
pip3 install tensorflow