研究室の共有PCの環境構築メモ
# 2024/08/06時点で未解決
書いた人: noka.icon
GOAL
python,torchからGPUにアクセスできるようcudaとかtorchのversionを揃える
- osはubunt
- GPUはRTX 2080
code:nvidia-smi
(base) kunitalab@kunitalab-System-Product-Name:~$ nvidia-smi
Tue Aug 6 17:13:32 2024
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 530.41.03 Driver Version: 530.41.03 CUDA Version: 12.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 NVIDIA GeForce RTX 2080 Off| 00000000:65:00.0 On | N/A |
| 24% 32C P8 30W / 215W| 135MiB / 8192MiB | 2% Default |
| | | N/A |
+-----------------------------------------+----------------------+----------------------+
+---------------------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=======================================================================================|
| 0 N/A N/A 1169 G /usr/lib/xorg/Xorg 56MiB |
| 0 N/A N/A 1442 G /usr/bin/gnome-shell 76MiB |
+---------------------------------------------------------------------------------------+
歴代のだれかかすでにいじった形跡あり
pipとconda環境が混ざっている
pipのほうを削除
公式をみて、
code:command
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
を実行した
torchは入ったが、
code:terminal
>> torch.__version__
'2.4.0'
>> torch.cuda.is_available()
False
>> torch.version.cuda
>>
GPUを認識していない。