Установка tf-nightly-gpu-2.0-preview в Google Colab - PullRequest
0 голосов
/ 16 января 2019

Я хотел бы протестировать новый предварительный просмотр TensorFlow 2.0 на Colab с поддержкой графического процессора, но после установки TensorFlow с использованием !pip install tf-nightly-gpu-2.0-preview в ячейку при импорте пакета появляется сообщение об ошибке: ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory что указывает на некоторое смещение с предустановленной версией CUDA на платформе.

1 Ответ

0 голосов
/ 16 января 2019

Решение состоит в том, чтобы переустановить CUDA 10.0 на Colab, следуя инструкциям:

!wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64 -O cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
!dpkg -i cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
!apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
!apt-get update
!apt-get install cuda
!pip install tf-nightly-gpu-2.0-preview

После выполнения предыдущих команд в ячейке вы можете импортировать и протестировать предварительный просмотр TensorFlow 2.0 с поддержкой графического процессора:

import tensorflow as tf
print(tf.__version__)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...