工商营业执照注册网站友情链接论坛

以下是Linux系统安装GPU版本PyTorch的完整步骤指南：

1. ‌环境检查与准备‌
确认系统架构‌
bash
Copy Code
uname -m  # 需为x86_64架构
cat /etc/os-release  # 查看Linux发行版信息:ml-citation{ref="3,4" data="citationList"}

检查GPU驱动‌
bash
Copy Code
nvidia-smi  # 验证驱动安装，显示GPU及支持的CUDA最高版本（需≥11.8）:ml-citation{ref="1,2" data="citationList"}

若无输出，需先安装NVIDIA驱动

2. ‌安装CUDA Toolkit（需与PyTorch版本匹配）‌

下载CUDA 11.8‌（PyTorch 2.0.1官方推荐）：

bash
Copy Code
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run

安装时‌取消勾选Driver‌（避免覆盖现有驱动）

配置环境变量‌：

bash
Copy Code
echo 'export PATH=/usr/local/cuda-11.8/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

验证：nvcc -V 应显示CUDA 11.8

4. ‌创建Python虚拟环境‌
bash
Copy Code
conda create -n pytorch_gpu python=3.9  # Python≥3.8
conda activate pytorch_gpu

5. ‌安装PyTorch GPU版‌

国内镜像加速‌（清华源）：
bash
Copy Code

pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 -f https://mirrors.tuna.tsinghua.edu.cn/pytorch/whl/torch_stable.html
 

6. ‌验证安装‌
python
Copy Code
import torch
print(torch.__version__)          # 应输出 2.0.1+cu118
print(torch.cuda.is_available())  # 应返回 True
print(torch.cuda.get_device_name(0))  # 显示显卡型号

⚠️ ‌常见问题解决‌

CUDA不可用‌：

检查驱动版本：nvidia-smi顶部版本需≥520.61.05
确认未安装CPU版PyTorch：pip uninstall torch后重装

安装超时‌：

bash
Copy Code
pip install ... --default-timeout=1000  # 延长超时时间:ml-citation{ref="8" data="citationList"}

版本冲突‌：

使用新虚拟环境隔离依赖
检查CUDA与PyTorch版本兼容性

完整环境依赖：

NVIDIA驱动 ≥520.61.05
CUDA Toolkit 11.8
cuDNN ≥8.x
Python 3.8~3.