Terminal 응용 conda 가상환경을 설정 한 뒤
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia conda install -c anaconda cudnn
export CUDA_VISIBLE_DEVICES=1 python main.py