溫馨提示×

PyTorch在CentOS上的安裝步驟有哪些

小樊
76
2025-04-08 23:26:03
欄目: 智能運維

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

1. 安裝Anaconda

  • 訪問Anaconda官網下載與CentOS匹配的Anaconda安裝包。
  • 安裝時勾選“Add Anaconda to PATH”。
  • 安裝完成后,打開終端輸入 conda --version,若顯示版本號,則安裝成功。

2. 創建虛擬環境

  • 創建虛擬環境命令:

    conda create -n pytorch_env python=3.8
    

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

  • 激活環境:

    conda activate pytorch_env
    

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

3. 安裝PyTorch

  • 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
    

4. 驗證安裝

  • 啟動Python解釋器:

    python
    
  • 在Python解釋器中輸入以下代碼驗證PyTorch安裝:

    import torch
    x = torch.rand(5, 3)
    print(x)
    

    如果輸出類似于 tensor([[0.3089, 0.5494, 0.1153], [0.9371, 0.8981, 0.3798], [0.3220, 0.4789, 0.6379], [0.3280, 0.0454, 0.5896], [0.2760, 0.0586, 0.2910]]) 的結果,則表示PyTorch安裝成功。

5. 使用國內鏡像源加速安裝(可選)

  • 如果安裝速度慢或失敗,可以使用國內鏡像源(如清華源):

    pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
    

    然后使用以下命令安裝PyTorch:

    pip install torch torchvision torchaudio
    

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

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