로딩...

cuda

설치

nvidia-smi
  • 명령어가 있는지 확인
  • 없는 경우 설치 진행
  • 사용 시스템과 아키텍쳐 os 를 고른후 network 선택하여 스크립트 실행

error

RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check

RuntimeError: Torch is not able to use GPU; add --skip-torch-cuda-test to COMMANDLINE_ARGS variable to disable this check
$ pip install torch torchvision torchaudio
  • 설치후
$ nvidia-smi
NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.
  • 이후에도 동일하게 에러발생
$ sudo apt-get install cuda-toolkit