diff --git a/setup_cuda.sh b/setup_cuda.sh index 2cfeed6..20181f8 100644 --- a/setup_cuda.sh +++ b/setup_cuda.sh @@ -60,7 +60,7 @@ apt-get update apt-get install -y --no-install-recommends cuda-11-2 - +#See for latest CUDA-Version: https://developer.nvidia.com/cudnn-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_network echo echo "Install libcudnn..." apt-get install -y libcudnn8_8.1.1.33-1+cuda11.2 libcudnn8-dev_8.1.1.33-1+cuda11.2