diff --git a/pytorch.spec b/pytorch.spec index c0a6295a6df1f2c94f390aad1684218ce055f3cf..3b04688a775d3596bc5d6b7cc8fd6ab7a28effa0 100644 --- a/pytorch.spec +++ b/pytorch.spec @@ -1,4 +1,4 @@ -%global anolis_version 7 +%global anolis_version 8 %global debug_package %{nil} Name: pytorch @@ -27,27 +27,21 @@ BuildRequires: qt5-qtbase-gui qt5-qtbase qt5-qtbase-common Requires: python3-numpy >= 1:1.19.4 Requires: python3-dataclasses +Requires: python3-typing-extensions -%if 0%{?fedora} >= 30 -BuildRequires: rocksdb-devel -%endif - -%if 0%{?rhel} BuildRequires: python3-dataclasses -%endif - %ifarch x86_64 BuildRequires: fbgemm-devel %endif -%global have_cuda 0 +%global have_cuda 1 %global have_tensorrt 0 %global vcu_maj 11 -%global vcu_min 5 -%global gpu_target_arch "3.5+PTX 6.1" +%global vcu_min 4 +%global gpu_target_arch "3.5+PTX 6.1 7.5 8.6" # % global gpu_target_arch "3.5+PTX;5.2;6.1;7.5;8.6" # % undefine _hardened_build @@ -58,11 +52,6 @@ BuildRequires: fbgemm-devel %undefine _missing_build_ids_terminate_build -%bcond_without cuda -%if %{without cuda} -%global have_cuda 0 -%endif - %if %{have_cuda} BuildRequires: cuda-nvcc-%{vcu_maj}-%{vcu_min} BuildRequires: cuda-nvtx-%{vcu_maj}-%{vcu_min} @@ -77,7 +66,7 @@ BuildRequires: libcurand-devel-%{vcu_maj}-%{vcu_min} BuildRequires: libcusparse-devel-%{vcu_maj}-%{vcu_min} BuildRequires: libcusolver-devel-%{vcu_maj}-%{vcu_min} BuildRequires: libnccl-devel -BuildRequires: libcudnn8-devel +BuildRequires: libcudnn-devel BuildRequires: gloo-devel BuildRequires: magma-devel Requires: cuda-cudart-%{vcu_maj}-%{vcu_min} @@ -290,14 +279,17 @@ export LD_LIBRARY_PATH="/usr/local/cuda-%{vcu_maj}.%{vcu_min}/%{_lib}/" -DUSE_GLOO=ON \ -DUSE_CUDA=ON \ -DUSE_CUDNN=ON \ - -DUSE_NVRTC=ON \ + -DUSE_NVRTC=OFF \ -DUSE_CUPTI_SO=ON \ -DUSE_FAST_NVCC=ON \ -DUSE_SYSTEM_NCCL=ON \ -DCMAKE_CUDA_FLAGS="-fPIC" \ -DCUDA_PROPAGATE_HOST_FLAGS=OFF \ -DTORCH_CUDA_ARCH_LIST=%{gpu_target_arch} \ + -DCUDA_TOOLKIT_ROOT_DIR="/usr/local/cuda-%{vcu_maj}.%{vcu_min}" \ -DCMAKE_CUDA_COMPILER="/usr/local/cuda-%{vcu_maj}.%{vcu_min}/bin/nvcc" \ + -DCUDA_NVCC_FLAGS="--compiler-options;-fPIC;-Wno-deprecated-gpu-targets;-allow-unsupported-compiler;--fatbin-options;-compress-all" \ + -DCMAKE_CUDA_FLAGS="--compiler-options -fPIC -Wno-deprecated-gpu-targets -allow-unsupported-compiler --fatbin-options -compress-all" \ -DNCCL_INCLUDE_DIR="%{_includedir}/nccl" \ -DUSE_MAGMA=ON \ -DBUILD_SPLIT_CUDA=ON \ @@ -313,9 +305,6 @@ export LD_LIBRARY_PATH="/usr/local/cuda-%{vcu_maj}.%{vcu_min}/%{_lib}/" -DUSE_ASAN=OFF \ -DUSE_ROCM=OFF \ -DUSE_MKLDNN=ON \ - -DUSE_NCCL=OFF \ - -DUSE_GLOO=OFF \ - -DUSE_CUDA=OFF \ %ifarch x86_64 -DUSE_FBGEMM=ON \ %else @@ -339,11 +328,7 @@ export LD_LIBRARY_PATH="/usr/local/cuda-%{vcu_maj}.%{vcu_min}/%{_lib}/" -DUSE_LMDB=ON \ -DUSE_REDIS=ON \ -DUSE_LEVELDB=ON \ -%if 0%{?fedora} >= 30 - -DUSE_ROCKSDB=ON \ -%else -DUSE_ROCKSDB=OFF \ -%endif -DUSE_FFMPEG=OFF \ -DUSE_OPENCV=OFF \ -DUSE_METAL=OFF \ @@ -439,7 +424,9 @@ rm -rf %{buildroot}/%{_includedir}/clog.h %{_libdir}/libtorch_cuda_cu.so %endif %{_libdir}/libnnapi_backend.so +%if !%{have_cuda} /usr/lib64/libcaffe2_detectron_ops.so* +%endif %{_libdir}/libshm.so %{_libdir}/libc10.so %{_libdir}/libtorch.so @@ -458,6 +445,9 @@ rm -rf %{buildroot}/%{_includedir}/clog.h %changelog +* Mon Jul 17 2023 Chunmei Xu - 1.10.1-8 +- rebuild with cuda enabled + * Mon Jul 17 2023 Chunmei Xu - 1.10.1-7 - add requires of python3-dataclasses