반응형
docker를 사용해보는중.
맥 -> 서버 ssh docker container 를 이용하여 파이참을 통해 디버깅해보고 싶음.
도커의 컨테이너에서 gpu를 사용하기위해 nvidia-container-toolkit을 설치하고
docker run --gpus all nvidia/cuda:10.0-base nvidia-smi
해봤지만 제목과 같은 오류만 발생..
이것저것 뻘짓을 다해보다가
서버 컴퓨터에 우분투를 설치하고나서 엔비디아 드라이버를 설치안했다는 것을 깨달음.
https://www.oofbird.me/55 를 참고하여 드라이버 설치후 다시 컨테이나를 런해봤음.
결과는 성공!
반응형
'딥러닝공부' 카테고리의 다른 글
mac, pycharm, docker container ssh (0) | 2021.07.30 |
---|---|
can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first. (0) | 2021.06.18 |
[pytorch] inference시 memory leak (0) | 2021.06.18 |
파이토치 에러 (0) | 2021.06.17 |
Prototypical Networks for Few-shot Learning _ 논문 리뷰_미완성 (0) | 2021.04.08 |
댓글