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.