docker pytorch 설치를 위한 gpu 설정

docker pytorch 설치를 위한 gpu 설정

작성일 2024.03.21댓글 1건
    게시물 수정 , 삭제는 로그인 필요

안녕하세요. 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로 원격연결하여 실행하는 방법을 알려주시면 감사하겠습니다.

++ 아래는 기존의 버전입니다. 
 


#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

profile_image 익명 작성일 -

1. python과 vscode 등 필요한 소프트웨어를 설치한 도커 이미지를 실행한 다음 pip 명령어로 pytorch를 설치하시면 되겠습니다.

2. 본인 gpu가 pytorch cuda 지원이 가능한지부터 한번 확인해보시기 바랍니다.

MX150 이면 노트북용 GPU일텐데

CUDA연산이 되는지 확인이 필요해보입니다.

cuda연산이 가능하더라도 nvidia cuda 소프트웨어를 설치해주어야 합니다.

3. docker 명령어 중 -exec가 있는데 이 명령어로 vscode 실행이 가능한지 한번 알아보시기 바랍니다.

M1 맥미니 로제타2 설치

... 도커는 docker preview 설치하셔야 합니다 m1용 docker가... 환경설정에서 글꼴 파일을 pptx, docx 파일 내에 함께... 반면에 맥북 프로 M1 은 발열관리를 위한 냉각팬이...

주피터노트북 GPU사용

... 주피터 노트북에서 GPU를 사용하려면, NVIDIA GPU 드라이버가 설치되어... 아래는 Windows에서 TensorFlow와 PyTorch를 사용하기 위한 GPU 설정 방법입니다. 1. NVIDIA...

딥러닝 개발환경 (colab 없이)

... 개발을 위한 훌륭한 도구입니다. NVIDIA GPU가 없는... 주로 사용되는 패키지로는 TensorFlow, PyTorch, Keras, scikit... 원하는 에디터를 설치하고 설정하세요. 데이터셋 준비...