ubuntu安装tensorflow-gpu
1.首先查看显卡驱动
nvidia-smi

可以看到:我的显卡驱动版本是390.77
2.根据显卡驱动版本确定cuda版本号
去nvidia网站NVIDIA CUDA Toolkit Release Notes查看显卡驱动版本与cuda版本的对应关系:

根据上面的对应关系,我的服务器是ubuntu14.04,因此选择cuda9.0
3.ubuntu一行命令在Anaconda下安装tensorflow-gpu
conda install -c anaconda tensorflow-gpu==1.9 cudatoolkit==9.0
等待安装完成就可以了。
博客介绍了在Ubuntu系统的Anaconda环境下安装tensorflow-gpu的方法。先查看显卡驱动版本,如390.77;再依据该版本在NVIDIA网站确定cuda版本,服务器为ubuntu14.04时选cuda9.0;最后用一行命令完成tensorflow-gpu的安装。

1490

被折叠的 条评论
为什么被折叠?



