set cuda version in Dockerfile.manylinux-cudaXXX

BaguaSys · Feb 22, 2023 · cfb49ec · cfb49ec
1 parent 12562b0
commit cfb49ec
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 0 deletions.
diff --git a/docker/Dockerfile.manylinux-cuda111 b/docker/Dockerfile.manylinux-cuda111
@@ -1,5 +1,8 @@
 FROM pytorch/manylinux-cuda111
 
+RUN rm -rf /usr/local/cuda
+RUN ln -s /usr/local/cuda-11.1 /usr/local/cuda
+
 RUN yum install centos-release-scl-rh -y && yum install devtoolset-8-toolchain -y
 RUN yum remove okay-release -y && yum install http://repo.okay.com.mx/centos/7/x86_64/release/okay-release-1-1.noarch.rpm -y
 RUN yum remove cmake3 -y && yum install cmake3 -y

diff --git a/docker/Dockerfile.manylinux-cuda113 b/docker/Dockerfile.manylinux-cuda113
@@ -1,5 +1,8 @@
 FROM pytorch/manylinux-cuda113
 
+RUN rm -rf /usr/local/cuda
+RUN ln -s /usr/local/cuda-11.3 /usr/local/cuda
+
 RUN yum install centos-release-scl-rh -y && yum install devtoolset-8-toolchain -y
 RUN yum remove okay-release -y && yum install http://repo.okay.com.mx/centos/7/x86_64/release/okay-release-1-1.noarch.rpm -y
 RUN yum remove cmake3 -y && yum install cmake3 -y

diff --git a/docker/Dockerfile.manylinux-cuda115 b/docker/Dockerfile.manylinux-cuda115
@@ -1,5 +1,9 @@
 FROM pytorch/manylinux-cuda115
 
+
+RUN rm -rf /usr/local/cuda
+RUN ln -s /usr/local/cuda-11.5 /usr/local/cuda
+
 RUN yum install centos-release-scl-rh -y && yum install devtoolset-8-toolchain -y
 RUN yum remove okay-release -y && yum install http://repo.okay.com.mx/centos/7/x86_64/release/okay-release-1-1.noarch.rpm -y
 RUN yum remove cmake3 -y && yum install cmake3 -y

diff --git a/docker/Dockerfile.manylinux-cuda116 b/docker/Dockerfile.manylinux-cuda116
@@ -1,5 +1,8 @@
 FROM pytorch/manylinux-cuda116
 
+RUN rm -rf /usr/local/cuda
+RUN ln -s /usr/local/cuda-11.6 /usr/local/cuda
+
 RUN yum install centos-release-scl-rh -y && yum install devtoolset-8-toolchain -y
 RUN yum remove okay-release -y && yum install http://repo.okay.com.mx/centos/7/x86_64/release/okay-release-1-1.noarch.rpm -y
 RUN yum remove cmake3 -y && yum install cmake3 -y