Win10下启动tensorflow程序加载显卡时间长

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/weixin_34613450/article/details/88310554

我是基于win10+tf1.12.0+cuda10配置的环境,但在第一次执行hello程序时加载资源很慢,去找了下资料,把步骤记载下来:

(tf-gpu) C:\Users\don> conda install -c anaconda cudatoolkit
(tf-gpu) C:\Users\don> conda install -c anaconda cudnn

在anaconda的对应环境中安装两个包cudatoolkit、cudnn,再次执行程序会很快

猜你喜欢

转载自blog.csdn.net/weixin_34613450/article/details/88310554