add a comment for GPU support

This commit is contained in:
Xu Yang
2024-07-15 10:08:20 +00:00
parent c4f5bd2f18
commit b4d89b3094
+3 -1
View File
@@ -1,4 +1,6 @@
FROM pytorch/pytorch:latest
FROM pytorch/pytorch:latest
# For GPU support, please choose the proper tag from https://hub.docker.com/r/pytorch/pytorch/tags
RUN apt-get clean && apt-get update && apt-get install -y \
curl \