sangeeta97

Setting cudatoolkit path can fix this, CUDNN_PATH=$(dirname $(python -c "import nvidia.cudnn;print(nvidia.cudnn.__file__)")) export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/:$CUDNN_PATH/lib export PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:32