这个tensorflow的gpu版本是真的难装啊,要自己安装cuda和cudnn,我试了几次也没装对,后来找到啦简单方法,一起来看看吧~
首先呢就是建立你自己的环境,我懒得搞就用的默认的 ,然后就是利用清华大学镜像,要不然太慢啊!!!
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
然后就是安装啦,这里需要注意,电脑也是要自动装cuda和cudnn的,默认安装的好像是cuda10.0,注意看一下电脑是否支持哈,不支持呢就安装版本低一点的tensorflow-gpu(conda install tensorflow-gpu=1.12.0)。耐心的等待安装!!!
这就是全过程啦~~~不过有人说anaconda安装的组件可能是不完整的,用别的库可能会报错,这个目前还没遇到,与到时候再说吧,反正目前tensorflow可以用啦!
版权声明:本文为m0_37723079原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。