# Pytorch 安裝
###### tags: blog文章

## Step 0. 建立虛擬環境
在開發Python專案時，常會遇到不同專案有不同的環境需求，而在管理這些不同的套件就會是一個問題。
如果只需要特定的套件，或是想要在不同的環境下測試，但又不想要開發環境受到影響，那虛擬環境是一個不錯的管理方法。
如果有閒的話，可以試試看docker X{D

以下我們使用Anaconda的套件管理系統作為示範。

conda create --name pytorch python=3.7  # pytorch 為你虛擬環境的名稱

接著跟著提示 啟動虛擬環境

conda activate pytorch

發現命令列前面多了一個(pytorch), 恭喜你成功進入虛擬環境。

## Step 1. 安裝pytorch
從官網的安裝連結得到適合自己環境的選擇
https://pytorch.org/get-started/locally/

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia  # 這邊以助教的電腦為例

## Step 2. 測試是否安裝成功

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

應該會得到類似的輸出:

tensor([[0.1252, 0.0325, 0.3591],
        [0.2043, 0.8560, 0.9773],
        [0.9602, 0.1425, 0.9884],
        [0.0791, 0.3461, 0.0628],
        [0.8484, 0.7485, 0.5202]])

如果裝的是GPU的版本, 可以嘗試以下指令測試CUDA driver能否使用:

>>>import torch 
>>>torch.cuda.is_available()
True

如果以上都成功執行, 恭喜你可以進到深度學習的世界!