ubuntu, 여러 conda environment 에 다른 쿠다 버전 설치

기본적으로 다른 드라이버 설치 되어있고, 여러 쿠다 버전 쓰고 싶다는 가정함

여기서는 cuda 11 들이 설치되어있는 환경에 12를 추가할 것임

 

1.  CUDA 설치 

https://developer.nvidia.com/cuda-toolkit-archive

CUDA Toolkit Archive

위와 같은 설정으로 선택 후 wget 으로 runfile 다운로드.

 

2. 설치

sudo sh cuda_12.1.0_530.30.02_linux.run --silent --toolkit --toolkitpath=/usr/local/cuda-12.1

 

 

3. conda env 준비

conda create -n 환경_이름 python==3.10 # 맞는 python 버전 선택
conda activate 환경_이름

 

 

4. conda 활성화/비활성화 스크립트 생성

기존의 cuda 버전이 11.1 을 사용하고, 해당 environment 에서 12.1를 사용한다고 했을 때, 

다음과 같은 두개의 파일을 생성한다.

4.1 set_vars.sh

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

4.2 unset_vars.sh

export CUDA_HOME=/usr/local/cuda-11.1
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_HOME/lib64
export LD_LIBRARY_PATH=$CUDA_HOME/lib:$LD_LIBRARY_PATH

 

5. 각 환경의 디렉토리에 저장

mkdir -p ~/anaconda3/envs/환경_이름/etc/conda/activate.d
mkdir -p ~/anaconda3/envs/환경_이름/etc/conda/deactivate.d

cp set_vars.sh ~/anaconda3/envs/환경_이름/etc/conda/activate.d/set_vars.sh
cp unset_vars.sh ~/anaconda3/envs/환경_이름/etc/conda/deactivate.d/unset_vars.sh

 

 

 

 

 

. 레퍼런스

https://xoft.tistory.com/86

[Ubuntu] CUDA 설치 & Multi CUDA 설치

https://developer.nvidia.com/