解决pytorch安装问题,以及NVIDIA CUDA与显卡驱动对照表

现在pytorch都是官网一键复制安装,很简单。

但是新手还是容易出现以下两个问题,原因是当显卡驱动与cuda不匹配(极少数是pytorch版本问题):

>>> torch.cuda.is_available
<function is_available at 0x0000021667561678>

or

>>> nvcc -V
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'nvcc' is not defined

两个验证都不行。所以换驱动或者换cuda版本

解决pytorch安装问题,以及NVIDIA CUDA与显卡驱动对照表

我去换显卡驱动了,没回来就是解决了。