반응형
이번에 florence2가 나왔습니다.
이에 대한 환경을 구축하는데 flash attn이 필요한데, 또 cuda11.7 필요!
docker 빠르게 구축했습니다.
https://hub.docker.com/r/pytorch/pytorch/tags?page=2&page_size=&ordering=&name=
1) 도커풀
docker pull pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
https://hub.docker.com/r/pytorch/pytorch/tags?name=&ordering=&page=2&page_size=
hub.docker.com
pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
2) 도커런
docker run -itd --gpus all --restart always --name cuda11.7 -v /mnt/d/data:/usr/src/data pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel /bin/bash
반응형
댓글