ubuntu一行命令在Anaconda下安装tensorflow-gpu
程序员文章站
2022-04-22 21:49:23
...
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
等待安装完成就可以了。