Skip to content

Update CUDA to version 11.5.2 [12.3.x] #7691

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
cmsbuild merged 2 commits into from
Mar 17, 2022
Merged

Update CUDA to version 11.5.2 [12.3.x] #7691

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ef2f9a8
Update to CUDA 11.5.2
fwyzard Mar 6, 2022
1ff1772
Update cuDNN to v8.3.2.44 for CUDA 11.5
fwyzard Mar 7, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 5 additions & 2 deletions cuda.spec
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
### RPM external cuda 11.4.4
### RPM external cuda 11.5.2
## INITENV +PATH LD_LIBRARY_PATH %i/lib64

%define runpath_opts -m compute-sanitizer -m drivers -m nvvm
%define driversversion 470.82.01
%define driversversion 495.29.05

%ifarch x86_64
Source0: https://developer.download.nvidia.com/compute/cuda/%{realversion}/local_installers/%{n}_%{realversion}_%{driversversion}_linux.run
Expand Down Expand Up @@ -99,6 +99,9 @@ ln -sf libcuda.so.1 %{i}
mv %_builddir/build/drivers/libnvidia-ptxjitcompiler.so.%{driversversion} %{i}/drivers/
ln -sf libnvidia-ptxjitcompiler.so.%{driversversion} %{i}/drivers/libnvidia-ptxjitcompiler.so.1
ln -sf libnvidia-ptxjitcompiler.so.1 %{i}/drivers/libnvidia-ptxjitcompiler.so
cp %{i}/nvvm/lib64/libnvvm.so.4.0.0 %{i}/drivers/
ln -sf libnvvm.so.4.0.0 %{i}/drivers/libnvvm.so.4
ln -sf libnvvm.so.4 %{i}/drivers/libnvvm.so

%post
# let nvcc find its components when invoked from the command line
Expand Down
34 changes: 15 additions & 19 deletions cudnn.spec
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,22 +1,24 @@
### RPM external cudnn 8.2.2.26
### RPM external cudnn 8.3.2.44
## INITENV +PATH LD_LIBRARY_PATH %i/lib64

%define cudaver 11.4
%define cudaver 11.5
%define cudnnver_maj %(echo %{realversion} | cut -f1,2,3 -d.)

%ifarch x86_64
Source: https://developer.download.nvidia.com/compute/redist/cudnn/v%{cudnnver_maj}/cudnn-%{cudaver}-linux-x64-v%{realversion}.tgz
%endif
%ifarch ppc64le
Source: https://developer.download.nvidia.com/compute/redist/cudnn/v%{cudnnver_maj}/cudnn-%{cudaver}-linux-ppc64le-v%{realversion}.tgz
%endif
# NVIDIA uses sbsa for aarch64, and the standard architecture name for ppc64le and x86_64
%ifarch aarch64
Source: https://developer.download.nvidia.com/compute/redist/cudnn/v%{cudnnver_maj}/cudnn-%{cudaver}-linux-aarch64sbsa-v%{realversion}.tgz
%define nvarch sbsa
%else
%define nvarch %{_arch}
%endif

# cuDNN archive base name and unpacked name
%define archive cudnn-linux-%{nvarch}-%{realversion}_cuda%{cudaver}-archive

Source: https://developer.download.nvidia.com/compute/redist/cudnn/v%{cudnnver_maj}/local_installers/%{cudaver}/%{archive}.tar.xz
Requires: cuda

%prep
%setup -n cuda
%setup -n %{archive}

if [ "${CUDA_VERSION%.*}" != %{cudaver} ]; then
echo 'Incompatible CUDA version in cudnn.spec!'
Expand All @@ -26,12 +28,6 @@ fi
%build

%install
%ifarch ppc64le
rm -f %_builddir/cuda/targets/ppc64le-linux/lib/*.a
mv %_builddir/cuda/targets/ppc64le-linux/lib %{i}/lib64
mv %_builddir/cuda/targets/ppc64le-linux/* %{i}/
%else
rm -f %_builddir/cuda/lib64/*.a
mv %_builddir/cuda/* %{i}/
%endif

# onnxruntime is hardcoded to look for the cudnn libraries under .../lib64
mv %_builddir/%{archive}/lib %{i}/lib64
mv %_builddir/%{archive}/* %{i}/