We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 872032b commit cc74eb0Copy full SHA for cc74eb0
Dockerfile
@@ -1,7 +1,7 @@
1
# FROM pytorch/pytorch:2.5.1-cuda12.1-cudnn9-devel as compile_server
2
FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-devel as compile_server
3
4
-ARG CPU_INSTRUCT=AVX2
+ARG CPU_INSTRUCT=NATIVE
5
6
# 设置工作目录和 CUDA 路径
7
WORKDIR /workspace
0 commit comments