torch and cuda 설치

황동준·2021년 8월 23일
0

window 10 환경에서 cuda 설정하기
차례로 따라하면 됩니다. 그러고 나서 nvcc --version 해주면 어떤 그래픽 카드에 cuda가 설치되었는지 볼 수 있음.

cuda와 torch 버전 맞추는 법
cuda와 torch 버전이 맞지 않으면 Torch not compiled with CUDA enabled 에러가 발생합니다.

profile
부담없이 기록하기

0개의 댓글