PyTorch 0.3 Build from source with CUDA 10 - CMake Error: - CUDA_cublas_device_LIBRARY (ADVANCED)

I am trying to build PyTorch 0.3 from the source with CUDA 10. During the build process below error is printed.

The command I used to build CUDA_HOME=/usr/local/cuda-10.0 python setup.py install

CMake Error: The following variables are used in this project, but they are set to NOTFOUND.
Please set them or make sure they are set and tested correctly in the CMake files:
CUDA_cublas_device_LIBRARY (ADVANCED)
    linked by target "THC" in directory /home/atumay/research/pytorch/torch/lib/THC

Is this because PyTorch 0.3 is not compatible with CUDA 10?

Yes, most likely CUDA10 is not supported in PyTorch 0.3 as the latter was released in ~Oct 2017 and was using CUDA7.5 back then.
I would not recommend to use such ancient PyTorch releases and instead to update to the latest release.

Okay, thank you I’ll consider switching to the latest version.

1 Like