로컬 환경에서 VScode를 사용하고 있었는데, 내 GPU로 딥러닝을 돌려보고 싶으신 분들을 위한 글입니다.
저 또한 안 돼서 빡쳤던 일화가 있어 남기는 글입니다.
님들은 고통받지 마세요.
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)
# output : "cuda"
VScode에서 위 코드를 실행하였을 때, 결과가 cuda 가 나타나도록 하는 것이 목적입니다.
혹시.. 그럴 리는 없겠지만.. NVIDIA 그래픽 카드 아니면 진행 불가합니다.
설치하는 경로만 알아서 잘 선택하시면 됩니다.
저의 경우는 11.7 버전을 사용하였으며 pytoch 사용에도 전혀 문제 없었습니다.
설치 완료 이후, C 드라이브 > Program files> NVIDIA GPU Computing Toolkit > CUDA > ‘설치한 버전’ 까지의 폴더 및 파일이 생성되었다면 잘 하신 겁니다.
여기를 편의사 Toolkit 폴더라고 부르겠습니다.
저의 경우는 11.x 최신 버전을 다운로드 하였으며 pytoch 사용에도 전혀 문제 없었습니다.
여기서 몇몇 폴더의 경로를 기억해야 합니다!
Toolkit 폴더 내부의 ‘bin’ 폴더, ‘include’ 폴더, ‘extra’ > ‘CUPT > ‘lib64’ 폴더까지 총 3개의 폴더 경로를 기억해주세요!
이제 귀찮은 부분들은 거의 끝났습니다.
아직은 우리가 활용할 Python 파일은 설치하지 않았으니, 그냥 방법만 기억해주세요. 우측 영상처럼 간단히 변경할 거예요.
우측 슬라이드와 같이 Prompt 찾아서 실행한 화면이 동일하면 성공입니다. 앞에 (base) 붙어서 나타날 겁니다.
이제 진짜 마지막입니다!
이제 VScode 에서 cuda 사용이 가능하실 겁니다.