CUDA: Difference between revisions
(→Linux) |
|||
Line 6: | Line 6: | ||
* Iinstall the latest nvidia drivers from the standard repo, e.g. <code>nvidia-drivers-440</code><br> | * Iinstall the latest nvidia drivers from the standard repo, e.g. <code>nvidia-drivers-440</code><br> | ||
* Install [https://developer.nvidia.com/cuda-toolkit Cuda Toolkit] separately without the drivers.<br> | * Install [https://developer.nvidia.com/cuda-toolkit Cuda Toolkit] separately without the drivers.<br> | ||
** Use one of the deb install options. | |||
* You may also want to install the following: | * You may also want to install the following: | ||
** [https://developer.nvidia.com/rdp/cudnn-download cuDnn]<br> | ** [https://developer.nvidia.com/rdp/cudnn-download cuDnn]<br> |
Revision as of 11:51, 13 April 2020
Installation
Linux
- Iinstall the latest nvidia drivers from the standard repo, e.g.
nvidia-drivers-440
- Install Cuda Toolkit separately without the drivers.
- Use one of the deb install options.
- You may also want to install the following:
- cuDnn
- TensorRT
- cuDnn
For tensorflow and pytorch, you may need to add LD_LIBRARY_PATH=/usr/local/cuda/lib64
to your environment variables.
You can also do this in PyCharm.