Hello. First of all, thank you to the Lambda team for providing the Lambda stack package!
I would like to install the Cuda 11.1 on my Ubuntu 18.04 with RTX 3090 to run a PyTorch program with GPU.
I followed the instruction and first installed by:
LAMBDA_REPO=$(mktemp) && \
wget -O${LAMBDA_REPO} https://lambdalabs.com/static/misc/lambda-stack-repo.deb && \
sudo dpkg -i ${LAMBDA_REPO} && rm -f ${LAMBDA_REPO} && \
sudo apt-get update && sudo apt-get install -y lambda-stack-cuda
sudo reboot
and then
sudo apt-get update && sudo apt-get dist-upgrade
.
However, unfortunately, the installed Cuda version is still 10.0, and I would like to ask for help.
I appreciate your time.
Additional information:
python3 6s
Python 3.6.9 (default, Oct 8 2020, 12:12:24)
[GCC 8.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'1.4.0'
nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130
nvidia-smi
Thu Dec 10 09:33:53 2020
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 455.45.01 Driver Version: 455.45.01 CUDA Version: 11.1 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 GeForce RTX 3090 On | 00000000:01:00.0 On | N/A |
| 32% 34C P8 35W / 350W | 360MiB / 24265MiB | 31% Default |
| | | N/A |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 1260 G /usr/lib/xorg/Xorg 164MiB |
| 0 N/A N/A 1405 G /usr/bin/gnome-shell 41MiB |
| 0 N/A N/A 1902 G ...AAAAAAAA== --shared-files 150MiB |
+-----------------------------------------------------------------------------+