☰

CUDA가 사용 가능한지 확인하고 사용 가능한 경우 환경 변수를 설정하는 방법이 있습니까?

tag-icon • tag-icon bash
CUDA가 사용 가능한지 확인하고 사용 가능한 경우 환경 변수를 설정하는 방법이 있습니까?

기계 학습을 위한 Python 프로그램이 있습니다. 셸 스크립트를 사용해서 실행하고 싶기도 하고, 가끔 CUDA_VISIBLE_DEVICESNVIDIA GPU를 사용하도록 환경 변수를 설정하기도 합니다.

CUDA가 설치되지 않은 M1 MacBook에서 동일한 프로그램을 실행하고 싶습니다. CUDA 설치 여부에 따라 환경 변수를 설정하는 방법이 있습니까?

CUDA가 (Linux에서) 사용 가능한지 확인하는 일반적인 방법은 nvcc --versionCUDA가 유효한 출력을 반환하면 설치된 것입니다. 나는 다음과 같은 것을 생각하고 있습니다 :

if "nvcc --version":
    export CUDA_VISIBLE_DEVICES=0

답변1

Bash 스크립트에서 구문은 다음과 같습니다.

if nvcc --version
then
  export CUDA_VISIBLE_DEVICES=0
fi

또는 더 짧게:

nvcc --version && export CUDA_VISIBLE_DEVICES=0

두 가지 모두 nvcc --version출력의 유효성을 확인하지 않고 명령의 종료 상태를 테스트합니다.

관련 정보