activate cuDNN
@@ -56,6 +56,7 @@ function Main {
# Setup environment
echo "Using CUDA $cuda and Python $python"
ActivateCUDA $cuda
+ ActivateCuDNN "8.6" $cuda
ActivatePython $python
# Setup build environment variables
