优秀的编程知识分享平台

网站首页 > 技术文章 正文

windows10安装最新tensorflow-gpu指南(2条指令搞定)

nanyue 2024-08-08 19:10:20 技术文章 11 ℃

windows10安装最新tensorflow-gpu指南(不需要手动配置Cuda和Cudnn)

小技巧:使用cmd在线安装,很多情况是会因为网络有连接超时,导致安装失败。因此,在安装失败时,复制cmd的连接,去迅雷进行下载。把下载好的文件,复制到anaconda\pkgs路径下,即可。

CPU版本的Tensorflow安装过程,在此就不说了,网上也很多教程。

Tensorflow GPU版本的安装教程如下:

1、安装Anaconda,这个必须的啦,在此就不说这个了;

2、可先预先下载好tensorflow-gpu版本(也可不下,到时在线安装下载超时,复制链接去迅雷下载,按照上面的小技巧即可)

3、从开始菜单找到Anaconda Prompt,打开,输入以上命令,直接开始安装tensorflow-gpu的各种依赖,我们可以看到下载的依赖里面已经包含了cuda和cudnn,当前安装的版本是tensorflow-gpu-1.13+cuda10+cudnn7.3。(下载超时,看小技巧)

conda create -n tensorflow_gpu tensorflow-gpu

conda activate tensorflow_gpu

安装过程比较久,目前,我两台电脑都是使用这个方法安装的,只针对windows版本安装。

CSDN:https://blog.csdn.net/weixin_40437821/article/details/102899098

参考:https://zhuanlan.zhihu.com/p/59109217

最近发表
标签列表