5,323
edits
CUDA: Difference between revisions
→Linux
(→Linux) |
(→Linux) |
||
| Line 23: | Line 23: | ||
# Install NVIDIA driver | # Install NVIDIA driver | ||
sudo apt install -y nvidia-driver- | sudo apt install -y nvidia-driver-450 | ||
# Reboot. Check that GPUs are visible using the command: nvidia-smi | # Reboot. Check that GPUs are visible using the command: nvidia-smi | ||
| Line 38: | Line 38: | ||
</pre> | </pre> | ||
For tensorflow and pytorch, you may need to add <code>LD_LIBRARY_PATH=/usr/local/cuda/lib64</code> to your environment variables.<br> | 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> | ||