Tensorflow2.3 GPU版本的安装 Windows10
我这电脑的Nivdia驱动版本是11.0,我查到的教程里都是要求10.1,没办法,自己莽装一下吧。
使用pip直接安装Tensorflow-GPU
-
先创建一个conda虚拟环境
conda create -n tf23gpu python=3.7
安装完成界面是这个
-
激活新环境
tf23gpu
activate tf23gpu
-
安装
tensorflow-gpu
,豆瓣源比较快pip install tensorflow-gpu==2.3 -i https://pypi.doubanio.com/simple/
-
测试安装好的tf23gpu
第一条提示信息里有一个警告提示:无法加载dll文件“cudart64_101.dll”
第二条提示信息显示:如果本机没有GPU请忽略上一条提示
可是我本机是有GPU的,显然此时的tensorflow是无法使用GPU的,当然不死心的我还是要测试一下的。继续在cmd中输入命令print('GPU', tf.test.is_gpu_available())
,最后一行显示GPU False
,明摆着GPU不能用啊!
-
修复GPU可用性
上面的报错信息显示缺失文件,那我就给他补充 cudart64_101.dll 文件 。度盘 提取码:6666
解压后将 cudart64_101.dll 文件放入路径D:\Softwares\Anaconda\envs\tf23gpu
(这是我的路径,请对坐入号)再次测试:
重新进入python会话。
中间还有一大串提示信息,这里不管,只看最后的输出GPU True
,舒服了