Я пытаюсь установить версию Tensorflow GPU, и я застрял в этом. Я установил nvidia-cuda-toolkit, запустив
sudo apt install nvidia-cuda-toolkit
, и он загрузился нормально. Но я не могу найти этот libcudart.so
Please specify which gcc nvcc should use as the host compiler. [Default is /usr/bin/gcc]: /usr/bin/gcc
Please specify the Cuda SDK version you want to use, e.g. 7.0. [Leave empty to use system default]:
Please specify the location where CUDA toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]: /usr/local/cuda
Invalid path to CUDA toolkit. /usr/local/cuda/lib64/libcudart.so cannot be found
Как я могу решить эту проблему?
Не уверенный, если это - лучший способ, но у меня была та же проблема, и это помогло.
sudo ln -s /usr/local/cuda/lib64 /usr
Проверяют ссылку от /usr
с ls -l lib64
lib64 -> /usr/local/cuda/lib64
Кажется, экспортом неправильного пути.
Так, На терминальном типе: sudo ldconfig /usr/local/cuda/lib64
ldconfig создает необходимые ссылки и кэш к новым общим библиотекам, найденным в каталогах, указанных на командной строке
, Если это не работает попытка: export PATH=$PATH:/usr/local/cuda/bin
Это установит экологический путь.
Если Вы используете Ubuntu 16.04 или Ubuntu 18.04 и хотите получить TensorFlow с установленной поддержкой GPU, существует deb пакет для этого в репозитории Стопки Лямбды.
Можно установить репозиторий и пакет с этой строкой:
LAMBDA_REPO=$(mktemp) && \
wget -O${LAMBDA_REPO} https://lambdal.com/static/files/lambda-stack-repo.deb && \
sudo dpkg -i ${LAMBDA_REPO} && rm -f ${LAMBDA_REPO} && \
sudo apt-get update && sudo apt-get install -y lambda-stack-cuda
Что это делает:
lambda-stack-cuda
пакет.LD_LIBRARY_PATH
или PATH
поскольку общие библиотеки размещаются в каталоги это ld
уже проверки во время ссылки.