Как обновить Tensorflow до v1.3 (cuDNN и обновление CUDA)
Во-первых, ниже моя системная среда
- ОС: Ubuntu 16.04
- текущий tensorflow: 1.0.1
- текущий cuDNN: 5.1.5
- текущий CUDA: 8.0
Я хочу обновить tensorflow до 1,3, но tf 1.3 требует cuDNN 6.
Так обновление cuDNN необходимо, но я не могу найти путь, как обновить cuDNN и настроить tensorflow, когда я установил прежде.
0
задан WKIm
1 September 2017 в 20:28
поделиться
1 ответ
Я записал сценарий, который может привыкнуть к чистой установке требуемая версия cuDNN или обновить существующий. После того как Вы выполняете шаги на странице GitHub, можно изменить cuDNN версию на v6.0. После этого Вы не должны делать конфигурацию для TensorFlow, она начнет работать автоматически с cuDNN v6.0. Можно найти сценарий здесь .
Это - то, что делает сценарий:
#!/bin/bash
rm -f /usr/include/cudnn.h
rm -f /usr/lib/x86_64-linux-gnu/*libcudnn*
rm -f /usr/local/cuda-*/lib64/*libcudnn*
cp -P packages/cudnn/include/cudnn.h /usr/include
cp -P packages/cudnn/lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*
rm -rf packages/cudnn
Это выполняет операции установки автоматически для версии, которую Вы хотите.
0
ответ дан dnzzcn
2 November 2019 в 08:05
поделиться