温馨提示×

Debian上PyTorch的常用命令有哪些

debian

小樊

56

2025-09-03 11:15:29

栏目: 智能运维

以下是Debian上PyTorch的常用命令，涵盖安装、验证及环境管理：

1. 系统与依赖准备

更新系统包：
sudo apt update && sudo apt upgrade -y
安装Python和pip：
sudo apt install python3 python3-pip

2. PyTorch安装命令

CPU版本（无GPU）：
pip install torch torchvision torchaudio
GPU版本（需先安装CUDA）：
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
（cu117替换为实际CUDA版本，如cu102、cu120等）

Conda安装（推荐隔离环境）：

conda create -n pytorch_env python=3.9 -y  
conda activate pytorch_env  
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch  # 替换CUDA版本

3. 验证安装

检查版本及CUDA可用性：
python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

4. 环境管理（可选）

创建虚拟环境（隔离项目）：
python3 -m venv pytorch_env
source pytorch_env/bin/activate
退出虚拟环境：
deactivate

5. 其他常用操作

卸载PyTorch：
pip uninstall torch torchvision torchaudio
查看已安装包：
pip list

说明：安装时需根据系统CUDA版本选择对应命令，可通过nvcc --version查看CUDA版本。推荐使用虚拟环境隔离不同项目依赖。

0 赞

0 踩

最新问答

相关问答

相关标签

产品服务

地区划分

专题活动

帮助支持

关于我们

售后咨询

7*24小时在线电话：400-100-2938

7*24小时在线 QQ：800811969

关注亿速云

亿速云公众号

手机网站二维码