5,323
edits
CUDA: Difference between revisions
→Installation
| Line 27: | Line 27: | ||
# Reboot. Check that GPUs are visible using the command: nvidia-smi | # Reboot. Check that GPUs are visible using the command: nvidia-smi | ||
# Install development and runtime libraries | # Install development and runtime libraries | ||
sudo apt-get install --no-install-recommends \ | sudo apt-get install --no-install-recommends \ | ||
cuda libcudnn7 libcudnn7-dev | cuda libcudnn7 libcudnn7-dev | ||
| Line 35: | Line 35: | ||
</syntaxhighlight> | </syntaxhighlight> | ||
For | For TensorFlow and PyTorch, you may need to add <code>LD_LIBRARY_PATH=/usr/local/cuda/lib64</code> to your environment variables.<br> | ||
You can also do this in PyCharm.<br> | You can also do this in PyCharm.<br> | ||
[[File:Pycharm LD LIBRARY PATH config.png| 200x200px]] | [[File:Pycharm LD LIBRARY PATH config.png| 200x200px]] | ||