PyTorch安裝及配置


PyTorch 環境配置及安裝

1. 安裝 anaconda

2. 為 PyTorch 創建單獨的環境,便於管理。

  1. 打開 Anaconda Prompt

  2. 為 PyTorch 創建單獨的環境
    python 版本與 anaconda 中 python 版本相同即可。
    conda create -n pytorch python=3.8

    輸入 y 開始安裝需要的包並創建環境(如果出現了 ERROR 並且你此時正在使用代理,嘗試關閉代理)。
    創建成功
    創建成功。

  3. 激活環境並查看環境下的包
    conda activate pytorch
    pip list

  4. 如果沒有 pytorch 包,需要下載安裝。

3. PyTorch 安裝

  1. pytorch 首頁:https://pytorch.org/

    選擇相關參數,如果電腦沒有Nvidia顯卡或者只有集顯,CUDA 選擇 CPU 即可。然后運行下面的指令。

  2. 在 pytorch 環境下運行指令。

  3. 安裝檢驗
    命令行輸入 python 進入 python 環境,執行 import torch ,如果沒有報錯,說明安裝成功。
    輸入 torch.cuda.is_available() 檢查 GPU 是否可用, 返回 true 即表示可用。

至此,PyTorch 的安裝配置完成。


4. 在新建的 pytorch 環境中安裝 Jupyter

  1. 進入 pytorch 環境 conda activate pytorch

  2. 安裝需要的包 conda install nb_conda

  3. 屏幕上顯示 done 即安裝成功。

  4. 使用 Jupyter jupyter notebook


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM