[Ubuntu] CUDA 경로 설정

상황

 

1. NVIDIA/CUDA 이미지를 기반으로 Dockerfile 구성

2. Dockerfile로 필요한 도커 이미지 생성

2-1. openSSH설치하는 코드가 있어 vscode로 접근가능

3. 도커 컨테이너 생성

 

 

ls /usr/local

# bin cuda cuda-12 cuda-12.1 ...

서버로 접속하여 설치된 쿠다 확인 및 사용할 쿠다 경로 확인

 

 

export PATH=/usr/local/cuda-12.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH

~/ 경로의 .bashrc를 열어 환경변수(위 코드) 추가

* cuda-12.1은 글쓴이가 사용하고 있는 쿠다 버전이므로 환경에 맞게 수정

 

source ~/.bashrc

위 명령어로 수정한 .bashrc를 적용

 

그 후 nvcc로 적용된 CUDA 버전을 확인