在安装TensorFlow-GPU时需要对应TensorFlow-GPU版本与CUDA、CUDNN的版本对应。每次需要更改环境的时候,都要各种找网址。这里做一个网址整理:
1、安装GPU版本之前先检查电脑是否支持GPU版本
链接: https://pan.baidu.com/s/19Sf1WUizsMDhGm4ax86hWA?pwd=df12https://pan.baidu.com/s/19Sf1WUizsMDhGm4ax86hWA?pwd=df12 提取码: df12
双击exe,安装软件选择否。
1代表的是显卡驱动版本,2代表的是支持CUDA计算
2、查找要安装的gpu版本和对应cuda、cudnn之间的版本对应关系
https://www.tensorflow.org/install/source_windows
3、查看电脑显卡驱动版本是否符合对应CUDA版本要求
除了前面使用TechPowerUp GPU-Z 软件查看外还可以使用: cmd输入nvidia-smi
nvidia-smi显示不是内部或外部命令也不是可运行的程序
链接:Release Notes :: CUDA Toolkit Documentation
4、下载对应版本的显卡驱动
官方驱动 | NVIDIA
这个网址打开会比较慢,不过这里推荐大家使用这种方式下载。
这样做的好处是:可以备份驱动安装包。
理由:如果需要卸载cuda时不小心卸载掉伦理显卡驱动的话,还要重新下(自己踩过的坑)。备份的话就不怕了,所以要养成备份的好习惯。
坑点:
1、我也试过在百度上搜索“英伟达显卡驱动下载”,点击搜索出来的官方页面会存在打不开的情况
2、使用驱动精灵、驱动人生之类的下载显卡驱动会限速,跟百度网盘一样让开会员。不开会员也就82多k。
题外话,下载太慢可以安装一个IDM,应该是破解好的。如果大家下载没有破解的话可以去网上再重新找一个。
链接: https://pan.baidu.com/s/1pFEYiwEMFHvyNPjttpDH_g?pwd=m35dhttps://pan.baidu.com/s/1pFEYiwEMFHvyNPjttpDH_g?pwd=m35d提取码: m35d
5、下载对应版本的CUDA
CUDA Toolkit Archive | NVIDIA Developer
6、下载对应版本的CUDNN
cuDNN Archive | NVIDIA Developer(需要注册账号)
7、安装CUDA和CUDNN
Cuda和cuDNN安装教程(超级详细)_jhsignal的博客-CSDN博客_cudnn安装
8、CUDA与CUDNN卸载
cuda 和 cudnn 库的卸载与安装 - 知乎
9、conda 源
豆瓣:pip install tensorflow-gpu==2.3.0 -i https://pypi.douban.com/simple/
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)