5,323
edits
CUDA: Difference between revisions
→Linux
(→Linux) |
(→Linux) |
||
| Line 38: | Line 38: | ||
</pre> | </pre> | ||
;Notes | |||
* <code>nvidia-driver-450</code> seems buggy. Would not recommend. | |||
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> | ||