pythorch 安裝

冷亦蓝發表於2024-11-16

PyTorch安裝指南

1. 安裝Conda

1.1 下載Conda

  1. 訪問Conda官網
  2. 根據你的作業系統(Windows、macOS、Linux)選擇合適的安裝包。
  3. 點選下載按鈕,等待下載完成。

1.2 安裝Conda

  1. 開啟下載的安裝包。
  2. 按照安裝嚮導的提示進行安裝。
    • 在Windows上,通常是雙擊安裝包並按照提示點選“下一步”。
    • 在macOS和Linux上,通常是執行安裝指令碼(例如bash Anaconda3-2023.07-1-Linux-x86_64.sh)。
  3. 安裝完成後,開啟終端或命令提示符,輸入conda --version,確認Conda已成功安裝。

2. 安裝PyCharm

2.1 下載PyCharm

  1. 訪問PyCharm官網
  2. 根據你的需求選擇社群版(Community)或專業版(Professional)。
  3. 點選下載按鈕,等待下載完成。

2.2 安裝PyCharm

  1. 開啟下載的安裝包。
  2. 按照安裝嚮導的提示進行安裝。
    • 在Windows上,通常是雙擊安裝包並按照提示點選“下一步”。
    • 在macOS和Linux上,通常是執行安裝指令碼(例如sh pycharm-community-2023.2.tar.gz)。
  3. 安裝完成後,開啟PyCharm,確認安裝成功。

3. 在Conda中建立PyTorch環境

3.1 建立環境

  1. 開啟終端或命令提示符。
  2. 輸入以下命令建立一個新的Conda環境:
    conda create -n pytorch python=3.9
    
  3. 按照提示確認安裝,等待環境建立完成。

3.2 啟用環境

  1. 輸入以下命令啟用新建立的環境:
    conda activate pytorch
    
  2. 確認環境已啟用(命令提示符前會顯示環境名稱,例如(pytorch))。

4. 下載CUDA 12.4

4.1 下載CUDA

  1. 訪問NVIDIA官網
  2. 根據你的作業系統選擇合適的CUDA版本(12.4)。
  3. 點選下載按鈕,等待下載完成。

4.2 安裝CUDA

  1. 開啟下載的安裝包。
  2. 按照安裝嚮導的提示進行安裝。
    • 在Windows上,通常是雙擊安裝包並按照提示點選“下一步”。
    • 在macOS和Linux上,通常是執行安裝指令碼(例如sh cuda_12.4.0_520.61.05_linux.run)。
  3. 安裝完成後,開啟終端或命令提示符,輸入nvcc --version,確認CUDA已成功安裝。

5. 安裝PyTorch

5.1 獲取安裝命令

  1. 訪問PyTorch官網
  2. 在“Get Started”頁面,選擇合適的選項(例如Stable、Linux、Conda、Python、CUDA 12.4)。
  3. 複製生成的安裝命令。

5.2 安裝PyTorch

  1. 在啟用的Conda環境中,輸入以下命令安裝PyTorch:
    conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
    
  2. 按照提示確認安裝,等待安裝完成。

6. 檢驗PyTorch安裝

6.1 進入Python環境

  1. 在啟用的Conda環境中,輸入以下命令進入Python環境:
    python
    

6.2 匯入PyTorch並檢驗CUDA可用性

  1. 在Python環境中,輸入以下命令匯入PyTorch:
    import torch
    
  2. 輸入以下命令檢驗CUDA是否可用:
    torch.cuda.is_available()
    
  3. 如果返回True,則表示PyTorch已成功安裝並且CUDA可用。

6.3 退出Python環境

  1. 輸入以下命令退出Python環境:
    exit()
    

7. 安裝Jupyter

7.1 安裝nb_conda_kernels

  1. 在啟用的Conda環境中,輸入以下命令安裝nb_conda_kernels
    conda install nb_conda_kernels
    
  2. 按照提示確認安裝,等待安裝完成。

8. 安裝IPyKernel

8.1 安裝ipykernel

  1. 在啟用的Conda環境中,輸入以下命令安裝ipykernel
    conda install ipykernel
    
  2. 按照提示確認安裝,等待安裝完成。

9. 將該環境註冊為Jupyter Kernel

9.1 註冊環境

  1. 在啟用的Conda環境中,輸入以下命令將環境註冊為Jupyter Kernel:
    python -m ipykernel install --user --name=pytorch --display-name "Python (pytorch)"
    
  2. 確認命令執行成功,環境已註冊。

10. 啟動Jupyter Notebook

10.1 啟動Jupyter

  1. 在啟用的Conda環境中,輸入以下命令啟動Jupyter Notebook:
    jupyter notebook
    
  2. 瀏覽器會自動開啟Jupyter Notebook介面。

10.2 選擇環境

  1. 在Jupyter Notebook介面中,點選“New”按鈕。
  2. 在下拉選單中選擇“Python (pytorch)”,這將啟動一個使用pytorch環境的Jupyter Notebook。

10.3 確認環境

  1. 在新建的Notebook中,輸入以下命令確認環境:
    import sys
    print(sys.executable)
    
  2. 確認輸出路徑指向pytorch環境。

相關文章