Conda와 gitbash를 이용한 CUDA 설치법

sangwon An (Hanho)·2023년 1월 5일
0

CUDA

목록 보기
1/1

간단한 gitbash를 이용한 conda 설치법

Terminal 응용
conda 가상환경을 설정 한 뒤

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

conda install -c anaconda cudnn

CUDA GPU 한 개만 응용

export CUDA_VISIBLE_DEVICES=1
python main.py
profile
고통스러운 야근

0개의 댓글