How to check which cuda version my pytorch is using

7

Thank you for the answer!

Added this code in my script,

logging.warning(‘cuda version: {}’.format(torch.version.cuda))
import os
logging.warning(‘CUDA_PATH: {}’.format(os.environ[“CUDA_PATH”]))
logging.warning(‘CUDA_HOME: {}’.format(os.environ[“CUDA_HOME”]))

Then, the outputs are like below:
WARNING: cuda version: 10.2
WARNING: CUDA_PATH: /opt/NVIDIA/cuda-9.1
WARNING: CUDA_HOME: /opt/NVIDIA/cuda-9.1

It seems working fine but I am still curious what pytorch refers to for the correct CUDA version under the hood.