Skip to content
Snippets Groups Projects
Commit ee95323f authored by Grégoire Kubler's avatar Grégoire Kubler
Browse files

fix : cuda container toolkit install condition

parent 783e193e
No related branches found
No related tags found
No related merge requests found
......@@ -46,6 +46,15 @@ variables:
case $CI_PROJECT_NAME in
"aidge_backend_cuda" )
apt-get install -y -qq libcudnn8-dev
# install nvidia container toolkit for cuda repo releases
if [[ "$CI_JOB_NAME" ~= "release" ]]; then
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
fi
;;
"aidge_backend_opencv" )
apt-get install -y -qq libopencv-dev
......@@ -53,14 +62,3 @@ variables:
* )
;;
esac
# install nvidia container toolkit for cuda repo releases
- >
if $CI_JOB_NAME ~= "release" && $CI_PROJECT_NAME ~= "cuda" ; then
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
fi
- set +x
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment