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

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

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

>>> torch.cuda.is_available

or

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

两个验证都不行。所以换驱动或者换cuda版本。可以自己打开nvidia控制面板或对照下表查看。

解决pytorch安装问题,以及NVIDIA CUDA与显卡驱动对照表_第1张图片

解决pytorch安装问题,以及NVIDIA CUDA与显卡驱动对照表_第2张图片

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

安装cuda建议一键安装并且不要更改默认路径。安装后我遇到了nvidia控制面板消失的问题,只能一遍遍的重新安装驱动或cuda。

解决pytorch安装问题,以及NVIDIA CUDA与显卡驱动对照表_第3张图片

tensorflow目前不支持python3.7。

 

你可能感兴趣的:(解决pytorch安装问题,以及NVIDIA CUDA与显卡驱动对照表)