@@ -29,32 +29,34 @@ sudo apt-get install -y \
29
29
# ccache
30
30
$( dirname " $0 " ) /ccache.sh
31
31
32
- wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb
32
+ # parse version number from command line argument
33
+ VERSION_DOTTED=${1-12.6} && VERSION_DASHED=$( sed ' s/\./-/' <<< ${VERSION_DOTTED} )
34
+
35
+ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
33
36
sudo dpkg -i cuda-keyring_1.0-1_all.deb
34
37
35
38
sudo apt-get update
36
39
sudo apt-get install -y \
37
- cuda-command-line-tools-11-3 \
38
- cuda-compiler-11-3 \
39
- cuda-cupti-dev-11-3 \
40
- cuda-minimal-build-11-3 \
41
- cuda-nvml-dev-11-3 \
42
- cuda-nvtx-11-3 \
43
- libcufft-dev-11-3 \
44
- libcurand-dev-11-3 \
45
- libcusparse-dev-11-3
46
- sudo ln -s cuda-11.3 /usr/local/cuda
40
+ cuda-command-line-tools-${VERSION_DASHED} \
41
+ cuda-compiler-${VERSION_DASHED} \
42
+ cuda-cupti-dev-${VERSION_DASHED} \
43
+ cuda-minimal-build-${VERSION_DASHED} \
44
+ cuda-nvml-dev-${VERSION_DASHED} \
45
+ cuda-nvtx-${VERSION_DASHED} \
46
+ libcufft-dev-${VERSION_DASHED} \
47
+ libcurand-dev-${VERSION_DASHED} \
48
+ libcusparse-dev-${VERSION_DASHED}
49
+ sudo ln -s cuda-${VERSION_DOTTED} /usr/local/cuda
47
50
48
51
# if we run out of temporary storage in CI:
49
- # du -sh /usr/local/cuda-11.3
52
+ # du -sh /usr/local/cuda-${VERSION_DOTTED}
50
53
# echo "+++ REDUCING CUDA Toolkit install size +++"
51
- # sudo rm -rf /usr/local/cuda-11.3 /targets/x86_64-linux/lib/libcu{fft,pti,rand}_static.a
52
- # sudo rm -rf /usr/local/cuda-11.3 /targets/x86_64-linux/lib/libnvperf_host_static.a
53
- # du -sh /usr/local/cuda-11.3 /
54
+ # sudo rm -rf /usr/local/cuda-${VERSION_DOTTED} /targets/x86_64-linux/lib/libcu{fft,pti,rand}_static.a
55
+ # sudo rm -rf /usr/local/cuda-${VERSION_DOTTED} /targets/x86_64-linux/lib/libnvperf_host_static.a
56
+ # du -sh /usr/local/cuda-${VERSION_DOTTED} /
54
57
# df -h
55
58
56
59
# cmake-easyinstall
57
- #
58
60
sudo curl -L -o /usr/local/bin/cmake-easyinstall https://raw.githubusercontent.com/ax3l/cmake-easyinstall/main/cmake-easyinstall
59
61
sudo chmod a+x /usr/local/bin/cmake-easyinstall
60
62
export CEI_SUDO=" sudo"
0 commit comments