Search

03.(211123)Torch_주피터 노트북에서 Torch를 쓰기위한 Torch 설치하기(AI학습을 위한)

맨처음 모든 세팅을 종료 시키고 파이토치를 제거하자(남은 찌꺼기 제거)
conda uninstall pytorch pip uninstall torch pip uninstall torch # run this command twice
Plain Text
복사
그리고 아래 사이트를 가자
아래 이미지 처럼 각 버전에 맞춰서 명령어를 찾아보자
명령어를 주피터 노트북에 입력
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
위 과정은 아래의 오류들 즉 버전문제에 대한 해결책으로도 쓰인다.
uda error: out of memory cuda kernel errors might be asynchronously reported at some other api call,so the stacktrace below might be incorrect.
Python
복사
RuntimeError: CUDA error: no kernel image is available for execution on the deviceCUDA
Python
복사
질문은 yklovejesus@gmail.com 으로 해주세요!

안녕하세요

한국전자기술연구원 김영광입니다.
관련 기술 문의와 R&D 공동 연구 사업 관련 문의는 “glory@keti.re.kr”로 연락 부탁드립니다.

Hello

I'm Yeonggwang Kim from the Korea Electronics Research Institute.
For technical and business inquiries, please contact me at “glory@keti.re.kr”