タイトル : GPUマシンをUbuntuに python環境
更新日 : 2024-05-05
カテゴリ : プログラミング
ubuntu24には、pythonは3.12.3が入ってます
$ which python3
/usr/bin/python3
$ python3 --version
Python 3.12.3
$pyenv入れます
https://github.com/pyenv/pyenv-installer を参照して以下ですね。.bashrcの変更も行います
curl https://pyenv.run | bashtorch用をとりあえず作成
pyenv install 3.12.3
pyenv virtualenv 3.12.3 torch2024https://pytorch.org/ を参照して、以下でインストール。CUDAの11.8と12.1では12.1を選択です。
pip3 install torch torchvision torchaudiotorchでGPUを確認します
ソース
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.get_device_name())
print(torch.cuda.get_device_capability())実行結果
$ python check01.py
True
1
NVIDIA GeForce RTX 4060
(8, 9)
$