Official whl file (torch=1.0.1 with CUDA 10.0) ships with wrong torch.version.cuda

I recently downgraded from PyTorch 1.1 to 1.0.1 to get an old project running.

Ubuntu 18.04
GPU: Nvidia 2080
CUDA: V10.0.130 (from nvcc --version)

After downloading/installing PyTorch 1.0.1 from via

pip install torch==1.0.1 -f

and running torch.version.cuda, It returns 9.0.176.

I believe this is the reason why I’m getting CUDA errors in my project. How can I get torch.version.cuda to return the correct CUDA version (10.0)?

Thank you in advance :slight_smile:

I have solved the problem by downloading/installing a different .whl file via pip install

This comment from the official PyTorch Github has led me to the solution.