PyTorch安裝指南
1. 安裝Conda
1.1 下載Conda
- 訪問Conda官網。
- 根據你的作業系統(Windows、macOS、Linux)選擇合適的安裝包。
- 點選下載按鈕,等待下載完成。
1.2 安裝Conda
- 開啟下載的安裝包。
- 按照安裝嚮導的提示進行安裝。
- 在Windows上,通常是雙擊安裝包並按照提示點選“下一步”。
- 在macOS和Linux上,通常是執行安裝指令碼(例如
bash Anaconda3-2023.07-1-Linux-x86_64.sh
)。
- 安裝完成後,開啟終端或命令提示符,輸入
conda --version
,確認Conda已成功安裝。
2. 安裝PyCharm
2.1 下載PyCharm
- 訪問PyCharm官網。
- 根據你的需求選擇社群版(Community)或專業版(Professional)。
- 點選下載按鈕,等待下載完成。
2.2 安裝PyCharm
- 開啟下載的安裝包。
- 按照安裝嚮導的提示進行安裝。
- 在Windows上,通常是雙擊安裝包並按照提示點選“下一步”。
- 在macOS和Linux上,通常是執行安裝指令碼(例如
sh pycharm-community-2023.2.tar.gz
)。
- 安裝完成後,開啟PyCharm,確認安裝成功。
3. 在Conda中建立PyTorch環境
3.1 建立環境
- 開啟終端或命令提示符。
- 輸入以下命令建立一個新的Conda環境:
conda create -n pytorch python=3.9
- 按照提示確認安裝,等待環境建立完成。
3.2 啟用環境
- 輸入以下命令啟用新建立的環境:
conda activate pytorch
- 確認環境已啟用(命令提示符前會顯示環境名稱,例如
(pytorch)
)。
4. 下載CUDA 12.4
4.1 下載CUDA
- 訪問NVIDIA官網。
- 根據你的作業系統選擇合適的CUDA版本(12.4)。
- 點選下載按鈕,等待下載完成。
4.2 安裝CUDA
- 開啟下載的安裝包。
- 按照安裝嚮導的提示進行安裝。
- 在Windows上,通常是雙擊安裝包並按照提示點選“下一步”。
- 在macOS和Linux上,通常是執行安裝指令碼(例如
sh cuda_12.4.0_520.61.05_linux.run
)。
- 安裝完成後,開啟終端或命令提示符,輸入
nvcc --version
,確認CUDA已成功安裝。
5. 安裝PyTorch
5.1 獲取安裝命令
- 訪問PyTorch官網。
- 在“Get Started”頁面,選擇合適的選項(例如Stable、Linux、Conda、Python、CUDA 12.4)。
- 複製生成的安裝命令。
5.2 安裝PyTorch
- 在啟用的Conda環境中,輸入以下命令安裝PyTorch:
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
- 按照提示確認安裝,等待安裝完成。
6. 檢驗PyTorch安裝
6.1 進入Python環境
- 在啟用的Conda環境中,輸入以下命令進入Python環境:
python
6.2 匯入PyTorch並檢驗CUDA可用性
- 在Python環境中,輸入以下命令匯入PyTorch:
import torch
- 輸入以下命令檢驗CUDA是否可用:
torch.cuda.is_available()
- 如果返回
True
,則表示PyTorch已成功安裝並且CUDA可用。
6.3 退出Python環境
- 輸入以下命令退出Python環境:
exit()
7. 安裝Jupyter
7.1 安裝nb_conda_kernels
- 在啟用的Conda環境中,輸入以下命令安裝
nb_conda_kernels
:conda install nb_conda_kernels
- 按照提示確認安裝,等待安裝完成。
8. 安裝IPyKernel
8.1 安裝ipykernel
- 在啟用的Conda環境中,輸入以下命令安裝
ipykernel
:conda install ipykernel
- 按照提示確認安裝,等待安裝完成。
9. 將該環境註冊為Jupyter Kernel
9.1 註冊環境
- 在啟用的Conda環境中,輸入以下命令將環境註冊為Jupyter Kernel:
python -m ipykernel install --user --name=pytorch --display-name "Python (pytorch)"
- 確認命令執行成功,環境已註冊。
10. 啟動Jupyter Notebook
10.1 啟動Jupyter
- 在啟用的Conda環境中,輸入以下命令啟動Jupyter Notebook:
jupyter notebook
- 瀏覽器會自動開啟Jupyter Notebook介面。
10.2 選擇環境
- 在Jupyter Notebook介面中,點選“New”按鈕。
- 在下拉選單中選擇“Python (pytorch)”,這將啟動一個使用
pytorch
環境的Jupyter Notebook。
10.3 確認環境
- 在新建的Notebook中,輸入以下命令確認環境:
import sys print(sys.executable)
- 確認輸出路徑指向
pytorch
環境。