[AI人工智能] 二、安裝python虛擬環境 + pytorch

紀錄 python安裝虛擬環境的過程。
切記，版本一定要照 [AI人工智能] 一、Python環境安裝 這裡的步驟安裝，否則跑程式很有可能會出現未知錯誤喔!

由於學習跟專案會拆兩邊，所以我會建議安裝兩個環境，一個是 Anaconda 的，另一個是 Pycharm 的，步驟基本如下，就看各位要建幾個。

1. 新建一個 python 虛擬環境:

-- 新建一個名為 python38_cuda_conda 的 python環境，版本為3.8
conda create -n python38_cuda_conda python=3.8

-- 切換到 python38_cuda_conda
-- 補充一下: 使用 activate 後，後續的安裝都建議使用 pip 來安裝，pip 提供比較齊全。
conda activate python38_cuda_conda

-- 安裝 torch 可以參考以下路徑:
-- 1. https://pytorch.org/get-started/locally/
-- 2. https://pytorch.org/get-started/previous-versions/
-- 3. torch有分 CPU, GPU 兩種版本，如果有顯卡就可以使用GPU版本。

-- GPU版本(需先查詢是否有顯卡，以及對應的cuda版本)
cmd -> nvidia-smi
-- 安裝指令類似如下，詳細就可以參考
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

-- CPU版本(一行即可)
cmd -> pip install torch


-- 補充: 如果有裝錯，那就把環境移除重裝，ENVIRONMENT 自行修改為您的環境名稱
-- conda remove --name ENVIRONMENT --all

最後可以驗證一下是否有成功安裝好。

> python
> 
> import torch
> 
> print(torch.cuda.is_available())

如果要確認環境的實體位址

-- 確認目前環境的路徑
1. 開啟Anaconda prompt，輸入 where conda
2. conda activate XXX，再輸入 where conda

上一篇: [AI人工智能] 一、Python、Anaconda、Pycharm安裝

下一篇: [AI人工智能] 三、Aanconda, Pycharm 切換py環境