5,323
edits
CUDA: Difference between revisions
→Installation
| Line 28: | Line 28: | ||
* For machine learning, use Anaconda or Docker's CUDA since different versions of TensorFlow and PyTorch require different CUDA versions. | * For machine learning, use Anaconda or Docker's CUDA since different versions of TensorFlow and PyTorch require different CUDA versions. | ||
You may need to add <code>LD_LIBRARY_PATH | 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]] | ||