docker compose로 해보려했으나 yml 파일 다음이 무엇인지 알 수가 없었다...
그래서 --link옵션으로 설정했다.
docker run -itd --name pytorch2 --link a282cf764a14 -v /mnt/d/train_dataset:/workspace -p 8889:8888 --gpus all --restart=always pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel
nvcc -V 의 영광을 이곳에 남긴다.
pytorch버전과 cuda버전이 맞아야 실행이 된다. 버전 문제가 참 많다..
- pytorch 이미지 버전 pytorch/pytorch:1.9.1-cuda11.1-cudnn8-devel
- cuda 이미지 버전 11.1
버전이 맞는 것을 link해줘야 했다.
'IT > Docker' 카테고리의 다른 글
[Docker] link 후 jupyter notebook, -p(port) 옵션 (0) | 2021.11.29 |
---|---|
[Docker] nvidia-docker (0) | 2021.11.29 |
[Docker] Systemctl 오류 (0) | 2021.11.25 |
[Docker] Docker compose (0) | 2021.11.23 |
[Docker] CUDA Image 가져와서 딥러닝 환경 세팅까지 (0) | 2021.11.23 |