5,323
edits
CUDA: Difference between revisions
→Linux
(→Linux) |
|||
| Line 12: | Line 12: | ||
;Adapted from [https://www.tensorflow.org/install/gpu#install_cuda_with_apt Tensorflow: Install cuda with apt] | ;Adapted from [https://www.tensorflow.org/install/gpu#install_cuda_with_apt Tensorflow: Install cuda with apt] | ||
< | <syntaxhighlight lang="bash"> | ||
# Add NVIDIA package repositories | # Add NVIDIA package repositories | ||
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.243-1_amd64.deb | wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.1.243-1_amd64.deb | ||
| Line 37: | Line 37: | ||
libnvinfer-dev=6.0.1-1+cuda10.1 \ | libnvinfer-dev=6.0.1-1+cuda10.1 \ | ||
libnvinfer-plugin6=6.0.1-1+cuda10.1 | libnvinfer-plugin6=6.0.1-1+cuda10.1 | ||
</ | </syntaxhighlight> | ||
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> | ||