使用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/谷歌。

相关推荐
机器之心6 小时前
谁说Scaling Law到头了?新研究:每一步的微小提升会带来指数级增长
人工智能·openai
coder_pig8 小时前
🤔 试试 OpenAI 的最强编程模型 "GPT-5-Codex"?
chatgpt·openai·claude
新智元12 小时前
收手吧 GPT-5-Codex,外面全是 AI 编程智能体!
人工智能·openai
安思派Anspire13 小时前
创建完整的评估生命周期以构建高(一)
aigc·openai·agent
机器之心14 小时前
刚刚,OpenAI发布GPT-5-Codex:可独立工作超7小时,还能审查、重构大型项目
人工智能·openai
l12345sy15 小时前
Day24_【深度学习—广播机制】
人工智能·pytorch·深度学习·广播机制
新智元1 天前
起猛了!这个国家任命 AI 为「部长」:全球首个,手握实权,招标 100% 透明
人工智能·openai
新智元1 天前
马斯克深夜挥刀,Grok 幕后员工 1/3 失业!谷歌 AI 靠人肉堆起,血汗工厂曝光
人工智能·openai
机器之心1 天前
用光学生成图像,几乎0耗电,浙大校友一作研究登Nature
人工智能·openai
蒋星熠1 天前
如何在Anaconda中配置你的CUDA & Pytorch & cuNN环境(2025最新教程)
开发语言·人工智能·pytorch·python·深度学习·机器学习·ai