无论安装Tensorflow 还是Pytorch GPU版的关键是cuda的安装
,装好cuda就已经成功了九成。
文章目录
查看显卡驱动版本
桌面右键=>NVDIA控制面板=>帮助=>系统信息=>显示页面中的驱动程序版本
如果没有也没关系,只要保证你的电脑是NVDIA显卡,后续CUDA安装包中包含了显卡驱动
记住该版本,一方面驱动版本要满足以下要求
下载CUDA安装包
我们要安装TensorFlow 2.1.0 及更高版本,其他版本可参考。
另外conda进行安装的过程中最好是按照上述的版本号,填写所有信息,比如
conda install tensorflow-gpu==1.13.1
不然很有可能conda会报找不到对应按照包的错误。
另外Tensorflow1.*的cuda版本最新也是10.0且大多数与Windows不兼容,尤其是python27环境,所有Windows最好就是直接上tensorflow2+的版本
安装 CUDA 10.1,在官网下载对应的版本。
点选电脑对应的操作系统版本,进行下载