리눅스(Linux) 환경에서 cuda 버젼 바꾸는 방법

cuda 버젼 바꾸는 방법

RTX 3090 관련 cuda

RTX 2080 TI와 다르게 RTX 3090은 Cuda 버젼 11.0 ~ 만 사용이 가능하다. RTX 2080 TI 10.0 (11.0) ~ 전부 사용이 가능하지만 3090은 cuda 버젼이 11.0 이하로 떨어지면 버젼 매칭이 되지 않는다 따라서 11.0 이상만 사용이 가능한데 대부분은 어떤 Cuda 버젼을 사용해야할지 애매한 경우가 많다. 또한 딥러닝 github를 확인해보면 cuda 버젼을 굉장히 다양하게 사용하는 경우가 많은데 이럴때 마다 cuda 버젼을 바꿔줄 수 있어야 하기 때문에 이 포스팅을 올리게 되었다.

cuda 버젼 바꾸기

대부분 cuda 버젼을 바꾸려고 하는 사람들이라면 각자 서버마다 특정 cuda 버젼을 이미 깔았고 깔 줄 안다고 가정을 하고 진행하도록 하겠다. cuda 버젼은 nvidia driver와 다르게 다양한 버젼을 미리 받아도 된다. 따라서 11.0~11.6 까지 전부 받도록 하자.
그 후에 사용자마다 cuda 버젼을 설정해 줄 수 있는데 그 방법은 다음과 같다.
해당 계정을 sudo 권한을 부여한 후 들어간 후에 아래 코드를 작성한다.

sudo nano ~/.bashrc

그 후에 마지막 code에 아래와 같이 작성한다.

[template]
export PATH="/usr/local/cuda-{version}/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-{version}/lib64:$LD_LIBRARY_PATH"
[cuda 11.3]
export PATH="/usr/local/cuda-11.3/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH"

이렇게 작성한 후 nvcc -V를 해보자

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Mon_May__3_19:15:13_PDT_2021
Cuda compilation tools, release 11.3, V11.3.109
Build cuda_11.3.r11.3/compiler.29920130_0

good! 11.3 버젼으로 바꼈다 ㅎㅎ