欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ubuntu一行命令在Anaconda下安装tensorflow-gpu

程序员文章站 2022-04-22 21:49:23
...

1.首先查看显卡驱动

nvidia-smi

ubuntu一行命令在Anaconda下安装tensorflow-gpu
可以看到:我的显卡驱动版本是390.77

2.根据显卡驱动版本确定cuda版本号

去nvidia网站NVIDIA CUDA Toolkit Release Notes查看显卡驱动版本与cuda版本的对应关系:
ubuntu一行命令在Anaconda下安装tensorflow-gpu
根据上面的对应关系,我的服务器是ubuntu14.04,因此选择cuda9.0

3.ubuntu一行命令在Anaconda下安装tensorflow-gpu

conda install -c anaconda tensorflow-gpu==1.9  cudatoolkit==9.0

等待安装完成就可以了。