无论安装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,在官网下载对应的版本。

点选电脑对应的操作系统版本,进行下载


本文详细介绍了在Windows10环境下,如何配置Tensorflow 2.x和Pytorch 1.x的GPU版本,包括检查NVIDIA显卡驱动、安装CUDA、设置环境变量、下载cuDNN、安装anaconda、安装Tensorflow-gpu和Pytorch GPU版,以及解决ImportError: DLL load failed的问题。确保CUDA版本与显卡驱动匹配,并正确设置环境变量,是成功安装的关键。
最低0.47元/天 解锁文章

1034

被折叠的 条评论
为什么被折叠?



