docker pytorch 설치를 위한 gpu 설정
-
게시물 수정 , 삭제는 로그인 필요
안녕하세요. docker 환경에서 pytorch를 깔고 싶습니다.
제 컴퓨터는 windows11, gpu는 geforece MX150을 사용하고 있습니다.
WSL용 Nvidia GPU Drive를 설치하는데 문제가 있습니다.
1. https://developer.nvidia.com/cuda/wsl 이 사이트에서 제 컴퓨터의 사양을 검색하니, 다음과 같은 드라이버가 추천되어 설치해 보았습니다.
GeForce Game Ready 드라이버
버전:
551.76 WHQL
배포 날짜:
2024.3.5
2. 설치하고 재부팅 후 nvidia-smi 결과 다음 사진과 같이 에러가 납니다..
docker를 사용한다해도, cuda를 깔아야하나요?
docker에서 pull한 pytorch를 VS Code로 원격연결하여 실행하는 방법을 알려주시면 감사하겠습니다.
++ 아래는 기존의 버전입니다.
GeForce Game Ready 드라이버
버전: | 551.76 WHQL | |
배포 날짜: | 2024.3.5 |
++ 아래는 기존의 버전입니다.
#docker pytorch #docker pytorch/pytorch #docker pytorch images #docker pytorch hub #docker pytorch jupyter #docker pytorch install #docker pytorch cuda 12.2 #docker pytorch cuda 11.8 #docker pytorch python 3.8 #docker pytorch jupyter notebook