使用pip3安装PyTorch和PyG

使用python3pip3命令安装python依赖库。

shell 复制代码
# python3 -V
Python 3.12.3
# 
# pip3 -V 
pip 25.0.1 from /root/.pyenv/versions/3.12.3/lib/python3.12/site-packages/pip (python 3.12)
ini 复制代码
Usage:   
  pip3 install [options] <package> ...
  pip3 install [options] -r <requirements file> ... 
  
  pip3 uninstall [options] <package> ...
  pip3 uninstall [options] -r <requirements file> ... 

GPU驱动和Cuda工具

GPU规格:NVIDIA TU104GL [Tesla T4] 驱动和Cuda版本:NVIDIA-SMI 550.127.08 Driver Version: 550.127.08 CUDA Version: 12.4

shell 复制代码
# lspci | grep -i nvidia
00:07.0 3D controller: NVIDIA Corporation TU104GL [Tesla T4] (rev a1)
# 
# nvidia-smi
Thu Apr 10 12:13:49 2025       
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 550.127.08             Driver Version: 550.127.08     CUDA Version: 12.4     |
|-----------------------------------------+------------------------+----------------------+
| 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  Tesla T4                       On  |   00000000:00:07.0 Off |                    0 |
| N/A   29C    P8              9W /   70W |       1MiB /  15360MiB |      0%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+
                                                                                         
+-----------------------------------------------------------------------------------------+
| Processes:                                                                              |
|  GPU   GI   CI        PID   Type   Process name                              GPU Memory |
|        ID   ID                                                               Usage      |
|=========================================================================================|
|  No running processes found                                                             |
+-----------------------------------------------------------------------------------------+
# 
# nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Mar_28_02:18:24_PDT_2024
Cuda compilation tools, release 12.4, V12.4.131
Build cuda_12.4.r12.4/compiler.34097967_0
# 
# nvidia-smi --query-gpu=name,memory.total,utilization.gpu,temperature.gpu --format=csv
name, memory.total [MiB], utilization.gpu [%], temperature.gpu
Tesla T4, 15360 MiB, 0 %, 29

PyTorch安装

PyTorch , Tensors and Dynamic neural networks in Python with strong GPU acceleration

shell 复制代码
pip3 install torch torchvision 
torch-2.6.0

pip3 install -r pytorch/requirements.txt 
pip3 install scikit-learn tqdm gpytorch skorch 

PyG安装

PyG is the ultimate library for Graph Neural Networks Build graph learning pipelines with ease

shell 复制代码
pip3 install torch_geometric 
torch_geometric-2.6.1

# python3 -c "import torch; print(torch.__version__)" 
2.6.0+cu124
# python3 -c "import torch; print(torch.version.cuda)" 
12.4
# python3 -c "import torch; print(torch.cuda.device_count())" 
1

pip3 install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.6.0+cu124.html 

Python示例代码依赖

shell 复制代码
pip3 install -r pyg-examples/requirements.txt 

如何导出Python项目的库依赖文件requirements.txt,自行搜索DeepSeek、ChatGPT或Google/谷歌。

相关推荐
waynaqua10 小时前
FastAPI开发AI应用五:模型新增图片理解
python·openai·fastapi
Kusunoki_D11 小时前
PyTorch 环境配置
人工智能·pytorch·python
Virgil1391 天前
【TrOCR】模型预训练权重各个文件解读
人工智能·pytorch·计算机视觉·自然语言处理·ocr·transformer
集成显卡1 天前
使用 Google 开源 AI 工具 LangExtract 进行结构化信息抽取
python·google·openai
机器之心1 天前
究竟会花落谁家?DeepSeek最新大模型瞄准了下一代国产AI芯片
人工智能·openai
赵英英俊1 天前
Python day51
人工智能·pytorch·python
爱学习的小道长1 天前
神经网络中 标量求导和向量求导
pytorch·深度学习·神经网络
伊织code1 天前
PyTorch - Developer Notes
pytorch·微分·精度
机器之心2 天前
好莱坞特效师展示AI生成的中文科幻大片,成本只有330元
人工智能·openai
钢铁男儿2 天前
使用 TensorBoardX 实现 PyTorch 神经网络可视化:从入门到进阶
人工智能·pytorch·神经网络