5,323
edits
CUDA: Difference between revisions
→Linux
(→Linux) |
(→Linux) |
||
| Line 25: | Line 25: | ||
sudo apt install libcudnn8 libcudnn8-dev | sudo apt install libcudnn8 libcudnn8-dev | ||
</syntaxhighlight> | </syntaxhighlight> | ||
;Notes | ;Notes | ||
* For machine learning, I just have Anaconda install a compatible CUDA since different versions of | * For machine learning, I just have Anaconda install a compatible CUDA since different versions of TensorFlow and PyTorch require different CUDA versions. | ||
You may need to add <code>LD_LIBRARY_PATH=/usr/local/cuda/lib64</code> to your environment variables.<br> | You may need to add <code>LD_LIBRARY_PATH=/usr/local/cuda/lib64</code> to your environment variables.<br> | ||