Установка tf-nightly-gpu-2.0-preview в Google Colab

if(@eval("\$result = $expresion;")===FALSE){
  $result=0;
}

Не поймает деление на 0 ошибок.

0
задан Luca Massaron 16 January 2019 в 18:34
поделиться

1 ответ

Решение состоит в том, чтобы переустановить 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 с поддержкой GPU:

import tensorflow as tf
print(tf.__version__)
0
ответ дан Luca Massaron 16 January 2019 в 18:34
поделиться
Другие вопросы по тегам:

Похожие вопросы: