Use github action from marketplace to install cuda

This commit is contained in:
Jason Bosco 2023-06-16 17:30:14 -05:00
parent 7cac0f4f54
commit c95c75ac12

View File

@ -32,25 +32,12 @@ jobs:
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo update-alternatives --set c++ /usr/bin/g++
# Install CUDA
wget --no-verbose https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
bash cuda_11.8.0_520.61.05_linux.run
rm cuda_11.8.0_520.61.05_linux.run
# Install cudnn
tar -xf cudnn-linux-x86_64-8.9.2.26_cuda11-archive.tar.xz
mv cudnn-linux-x86_64-8.9.2.26_cuda11-archive cudnn
cp cudnn/include/cudnn*.h /usr/local/cuda/include
cp cudnn/lib/libcudnn* /usr/local/cuda/lib64
chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
rm -rf cudnn*
# Copy these to ~/.profile
echo "LD_LIBRARY_PATH=$(echo $LD_LIBRARY_PATH:/usr/local/cuda/lib64)" >> $GITHUB_ENV
echo "CUDA_HOME=/usr/local/cuda" >> $GITHUB_ENV
echo "PATH=$(echo /usr/local/cuda/bin:$PATH) >> $GITHUB_ENV
- name: Install cuda
uses: Jimver/cuda-toolkit@v0.2.10
id: cuda-toolkit
with:
cuda: '11.8.0'
- name: Set up Bazel
uses: bazelbuild/setup-bazelisk@v2