☰

tensorflow 2 gpu安装问题ubuntu 18.04

tag-icon • tag-icon 18.04 nvidia cuda tensorflow
tensorflow 2 gpu安装问题ubuntu 18.04

我想在新安装的 ubuntu 18.04 上安装支持 gpu 的 tensorflow 2。

安装 ubuntu18.04 后，我运行了 tensorflow 网站中提供的以下代码。

# Add NVIDIA package repositories here
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.243-1_amd64.deb

sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

sudo dpkg -i cuda-repo-ubuntu1804_10.1.243-1_amd64.deb

sudo apt-get update

wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb

sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb

sudo apt-get update

# Install NVIDIA driver
sudo apt-get install --no-install-recommends nvidia-driver-450

重新启动命令的输出后

nvidia-smi

2020 年 10 月 18 日 星期日 13:35:11       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.80.02 驱动程序版本：450.80.02 CUDA 版本：11.0 |
|-------------------------------+----------------------+----------------------+
| GPU 名称 Persistence-M| 总线 ID Disp.A | 易失性 Uncorr. ECC |
| 风扇温度性能功率：使用情况/容量| 内存使用情况| GPU-Util 计算 M。|
| | | MIG M. |
|================================+=========================+========================|
| 0 GeForce GTX 1060 关闭 | 00000000:01:00.0 关闭 | N/A |
| N/A 52C P0 26W / N/A | 222MiB / 6078MiB | 0% 默认 |
| | | 不适用|
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| 流程: |
| GPU GI CI PID 类型 进程名称 GPU 内存 |
| ID ID使用情况 |
|==================================================================================|
| 0 N/AN/A 1522 G /usr/lib/xorg/Xorg 135MiB |
| 0 N/AN/A 1666 G /usr/bin/gnome-shell 78MiB |
| 0 不适用 4021 G /usr/lib/firefox/firefox 1MiB |
| 0 不适用 4898 G /usr/lib/firefox/firefox 1MiB |
| 0 N/AN/A 4959 G /usr/lib/firefox/firefox 1MiB |
+-----------------------------------------------------------------------------+

 
dipanjan@msi:~$ sudo apt-get install --no-install-recommends \
> cuda-10-1 \
> libcudnn7=7.6.5.32-1+cuda10.1 \
> libcudnn7-dev=7.6.5.32-1+cuda10.1
[sudo] dipanjan 的密码：
正在读取软件包列表...完成
构建依赖关系树       
正在读取状态信息...完成
E: 无法找到软件包 cuda-10-1

错误是：无法找到包 cuda-10-1  
我曾尝试通过在线查找来解决这个问题，但都不起作用。  
请指导我。

相关内容