PyTorch and CUDA driver

前端 未结 3 1746
隐瞒了意图╮
隐瞒了意图╮ 2021-01-19 06:40

I have CUDA 9.2 installed. For example:

(base) c:\\>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built          


        
相关标签:
3条回答
  • 2021-01-19 06:58

    Ever since https://github.com/pytorch/pytorch/releases/tag/v0.3.1, PyTorch binary releases had removed support for old GPUs' with CUDA capability 3.0. According to https://en.wikipedia.org/wiki/CUDA, the compute capability of Quadro K3000M is 3.0.

    Therefore, you might have to build pytorch from source or try other packages. Please refer to this thread for more information -- https://discuss.pytorch.org/t/pytorch-no-longer-supports-this-gpu-because-it-is-too-old/13803.

    0 讨论(0)
  • 2021-01-19 07:01

    I had a similar problem, you have to check on NVIDIA control panel that your card is selected by default.

    0 讨论(0)
  • 2021-01-19 07:18

    PyTorch official call for using CUDA 9.0 and I would suggest the same. In other cases, there are sometimes build issues which leads to 'CUDA not detected'.So, when using PyTorch its best to use CUDA 9.0 and CuDnn 7. I'll add a link where you can easily install Cuda 9.0 and CuDnn 7.

    https://yangcha.github.io/CUDA90/

    0 讨论(0)
提交回复
热议问题