码迷,mamicode.com
首页 > 系统相关 > 详细

Ubuntu 20.04安装tensorflow GPU版本(NVIDIA GTX-1060)

时间:2021-02-22 12:22:33      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:mode   volatil   cuda   ecc   UNC   vol   red   mem   arc   

1 安装nvidia驱动

$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001C03sv00001B4Csd000011D7bc03sc00i00
vendor   : NVIDIA Corporation
model    : GP106 [GeForce GTX 1060 6GB]
driver   : nvidia-driver-450-server - distro non-free
driver   : nvidia-driver-450 - distro non-free
driver   : nvidia-driver-390 - distro non-free
driver   : nvidia-driver-460 - distro non-free recommended
driver   : nvidia-driver-418-server - distro non-free
driver   : xserver-xorg-video-nouveau - distro free builtin

安装指定版本的驱动,一般安装推荐的版本(recommended)即可,我此处安装的是450版本。
sudo apt install nvidia-driver-450

安装后重启
sudo reboot

进入系统后,输入nvidia-smi查看当前GPU的基础信息,确认该版本驱动是否安装成功

$ nvidia-smi
Sun Feb 21 16:58:51 2021       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.102.04   Driver Version: 450.102.04   CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  GeForce GTX 106...  Off  | 00000000:01:00.0  On |                  N/A |
|  0%   57C    P8    10W / 120W |    567MiB /  6075MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A       875      G   /usr/lib/xorg/Xorg                194MiB |
|    0   N/A  N/A      1188      G   /usr/bin/kwin_x11                 116MiB |
|    0   N/A  N/A      1190      G   /usr/bin/plasmashell               41MiB |
|    0   N/A  N/A      1492      G   /usr/bin/plasma-discover           16MiB |
|    0   N/A  N/A      3595      G   /usr/lib/firefox/firefox            1MiB |
|    0   N/A  N/A      3719      G   /usr/lib/firefox/firefox            1MiB |
|    0   N/A  N/A      4053      G   ...gAAAAAAAAA --shared-files      188MiB |
+-----------------------------------------------------------------------------+

2 安装CUDA

存档见https://developer.nvidia.com/cuda-toolkit-archive

参考资料:
https://towardsdatascience.com/installing-tensorflow-gpu-in-ubuntu-20-04-4ee3ca4cb75d
https://cyfeng.science/2020/05/02/ubuntu-install-nvidia-driver-cuda-cudnn-suits/

Ubuntu 20.04安装tensorflow GPU版本(NVIDIA GTX-1060)

标签:mode   volatil   cuda   ecc   UNC   vol   red   mem   arc   

原文地址:https://www.cnblogs.com/codeRhythm/p/14424149.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有
迷上了代码!