updated the script for building wheels

bytedance · Jan 16, 2024 · df70a65 · df70a65
1 parent 8abe3ea
commit df70a65
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dockerfiles/manylinux/build_wheels.sh b/dockerfiles/manylinux/build_wheels.sh
@@ -13,6 +13,8 @@ export CUDA_HOME=/usr/local/cuda
 export CUTENSOR_DIR=/usr/local/cuda
 export PATH=$CUDA_HOME/bin:$PATH
 export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
+
+# blas is required by DFTD3 and DFTD4
 yum install openblas-devel
 
 # Compile wheels