现在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版本。可以自己打开nvidia控制面板或对照下表查看。
我去换显卡驱动了,没回来就是解决了。
安装cuda建议一键安装并且不要更改默认路径。安装后我遇到了nvidia控制面板消失的问题,只能一遍遍的重新安装驱动或cuda。
tensorflow目前不支持python3.7。