TensorFlow2.0-安装个东西怎么就那么难呢
文章目录
windows下安装Tensorflwo2.0
- 使用python3.7
硬件支持
保证Nvidia显卡的算力至少在3.0以上(含3.0)
查看Nvidia显卡的算力
-
cmd命令行查看
nvidia-smi
。程序的默认位置在C:\Program Files\NVIDIA Corporation\NVSMI
下的nvidia-smi.exe
。- 为了时能在命令行中使用改命令,需要将
C:\Program Files\NVIDIA Corporation\NVSMI
加入系统的环境变量中。
- 为了时能在命令行中使用改命令,需要将
软件支持
1.NVIDIA驱动程序
- 要求使用410.x或者更高版本。
- 低版本需要到官网下载新的驱动包。但是官网下载又太慢,那么就可以使用迅雷:https://blog.****.net/weixin_37044551/article/details/90487906。
- 也可以在安装
GeForce Experience软件,会自动询问是否保持更新。
- 鲁大师可以检测新的驱动,并进行安装。
- 无论哪种方式,都要确保驱动版本符合条件。
2.CUDA-10.0版本
conda install cudatookkit=10.0.130
3.cudnn-版本>7.4.1
conda install cudnn=7.6.0
安装Tensorflow2.0-gpu版本
-
在确保上面的python语言版本,NVIDIA显卡算力,NVIDIA驱动程序,CUDD版本,cudnn版本均正确后,执行
pip install tensorflow-gpu==2.0.0
-
一开始常使用
conda install tensorflow-gpu==2.0.0
但是显示dynamic library cudart64_100.dll
;链接失败,所以选择自己制定各个软件的版本号,然后采用pip
的方式安装,结果没有问题。
测试
执行
tf.test.is_gpu_available()
结果返回说明安装正确。【上述的执行需要一定时间,需要耐心等待。】
其他:
- 保证NVIDIA的驱动在410.x以上
- 执行
conda install tensorflow-gpu==2.0.0
- 但是这个操作(第一步成功,第二步太慢了),个人尝试出现了问题,所以采用了上述原生的方式进行安装。