溫馨提示×

PyTorch在Linux上的安裝步驟

小樊
56
2025-04-13 17:24:42
欄目: 智能運維

在Linux上安裝PyTorch的步驟如下:

安裝Anaconda

  1. 訪問Anaconda官網下載與操作系統匹配的版本安裝包。
  2. 安裝時勾選“Add Anaconda to PATH”(Windows用戶必須勾選,否則后續命令無法直接使用)。
  3. 安裝完成后,打開終端輸入 conda --version,若顯示版本號(如 conda 23.7.4),則安裝成功。

創建虛擬環境

  1. 創建虛擬環境命令:
conda create -n pytorch_env python=3.8

Python 3.8是兼容PyTorch的穩定版本,可替換為3.9或3.10(需確認PyTorch支持)。

  1. 激活環境:
conda activate pytorch_env

激活后終端提示符應顯示 (pytorch_env)。

安裝PyTorch和CUDA(如果使用GPU)

CPU版本

conda install pytorch torchvision torchaudio cpuonly -c pytorch

GPU版本(需NVIDIA顯卡和CUDA 11.8)

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

驗證安裝

  1. 查看自己的CUDA型號,輸入如下命令:
nvidia-smi
  1. PyTorch官網找到合適的版本,復制相應的命令到終端進行安裝。

驗證PyTorch是否安裝成功

在創建的小環境下輸入 python,在Python解釋器中輸入以下代碼,返回 True,安裝成功:

import torch
torch.cuda.is_available()

以上步驟應該可以幫助你在Linux系統上成功安裝PyTorch。如果在執行過程中遇到問題,可以參考PyTorch的官方文檔或相關社區論壇尋求幫助。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女