So I followed the steps to build from source, @ptrblck. All fine. Something I noticed after the build finished is that now I have in my anaconda env Pytorch 1.2 installed… Anyway, I’m now getting this error:
RuntimeError: cuDNN error: CUDNN_STATUS_MAPPING_ERROR.
Could CUDA version I have installed have something to do with this?, I have CUDA 10.2. I’m wondering because there is no mention about this version in the steps, but instead CUDA 10.0 or 10.1. When I installed the dependencies for the build, I ran conda install -c pytorch magma-cuda102 since my CUDA version is 10.2. But in the steps there is only mention about magma-cuda90 | magma-cuda92 | magma-cuda100 | magma-cuda101.
Also while googling around I found some extra env variables I did not set, such as:
export CUDA_NVCC_EXECUTABLE="/usr/local/cuda-x.y/bin/nvcc"
export CUDA_HOME="/usr/local/cuda-x.y"
export CUDNN_INCLUDE_PATH="/usr/local/cuda-x.y/include/"
export CUDNN_LIBRARY_PATH="/usr/local/cuda-x.y/lib64/"
export LIBRARY_PATH="/usr/local/cuda-x.y/lib64"

# and also
export USE_CUDA=1 USE_CUDNN=1 USE_MKLDNN=1

where x.y is CUDA version. They aren’t mentioned in the steps provided by the repo. Are these important? I didn’t get any error and all compiled just fine.
Thanks a lot.