Anaconda 환경설정 (서버에서 설정시)

nvcc -V으로 cuda 버전을 확인한다.

• 이 때 그래픽 카드에 맞는 cuda 버전인지 확인하는 게 필수!
• bashrc파일을 열고 cuda 경로 설정해줘야함!

nano ~/.bashrc
//맨아래 위 두 문장을 추가 시켜줍니다. 다만 cuda 버전에 따라서 버전을 다르게 입력
//cuda 버전이 10.1인경우는 cuda-10-1로 변경
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH

// CUDA 경로 업데이트
source ~/.bashrc
nvcc -V

conda 가상환경 생성, 삭제

• 생성

conda create -n [가상환경이름] python=3.x

• 삭제

conda remove --name [가상환경이름] --all

Torch와 Cuda 연결

• Cuda 버전과 호환되는 Torch를 설치.

#아래 코드를 이용해서 cuda와 Pytorch가 연결 잘 되있는지 확인!
import torch
torch.cuda.get_device_name(0)
torch.cuda.is_available()