使用命令創建一個新的虛擬環境
conda create --name XX python=3.8 -y
啟動這個虛擬環境
conda activate XX
安裝必要的工具包
pip install torch torchvision
pip install matplotlib
將所創建的虛擬環境加入到jupyter notebook的kernel中
conda install -c anaconda ipykernel
python -m ipykernel install --user --name=XX
啟動jupyter notebook
jupyter notebook
工具:pytorch,torchvision,matplotlib
tensorflow和pytorch區別:
tensorflow基於靜態圖,需要對計算過程/計算圖先構建好,然後再將輸入送入計算圖。
pytorch基於動態圖,就是說pytorch每次前向計算的時候會重現構建一個新的計算圖,這個特性使pytorch更加直觀,寫代碼的邏輯更加接近python本身的風格,便於調試。