解决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版本
我去换显卡驱动了,没回来就是解决了。