We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c96fc69 commit 21c69d1Copy full SHA for 21c69d1
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=NATIVE
+ARG CPU_INSTRUCT=AVX2
5
6
# 设置工作目录和 CUDA 路径
7
WORKDIR /workspace
0 commit comments