PyTorch 安裝指南
本博客主要介紹在 Anaconda虛擬環境中創建 PyTorch環境,默認已經安裝成功 Anaconda軟體,
Anaconda Individual Edition的官方下載鏈接:
https://www.anaconda.com/products/individual
按照自己的作業系統,選擇需要的版本號,隨后傻瓜式安裝即可,具體的安裝及使用可以參考以下鏈接:
https://blog.csdn.net/ITLearnHall/article/details/81708148
創建PyTorch的虛擬環境
在 【最近添加】 里面找到 Anaconda Prompt并打開,依次輸入以下命令:
# 創建名為 PyTorch_envs 的虛擬環境,并指定 Python 版本號為 python 3.8
conda create -n PyTorch_envs python=3.8
# 激活虛擬環境 PyTorch_envs
conda activate PyTorch_envs
# 退出虛擬環境
conda deactivate
虛擬環境中安裝 PyTorch
打開PyTorch官網,選擇安裝,【這里我們選擇官網安裝方式,其他的安裝方式博主也嘗試過,但是效果都不是很好,所以說最好還是按照官網的安裝方式進行,畢竟羊毛出在羊身上嘛,】

按照官網的提示安裝 PyTorch,這里需要注意的是安裝之前必須知道自己CUDA版本號,這里可以參考鏈接
https://blog.csdn.net/qq_38295511/article/details/89223169

我自己的CUDA版本號是10.1,所以選擇如圖示,
安裝方式建議選擇pip方式,作業系統根據自己的作業系統進行選擇,這里不再進行贅述,【最好選擇相對好一點的網路環境】如果下載速度非常慢的話,建議換源安裝,注意指明版本號,具體的換源方式可以參考:
https://blog.csdn.net/Ginomica_xyx/article/details/109605629

驗證 PyTorch 是否安裝成功
官網 https://pytorch.org/get-started/locally/#windows-verification 也說明了驗證方式:


如圖所示,PyTorch安裝成功,
安裝程序中可能出現的問題
import torch
出現
ImportError: numpy.core.multiarray failed to import
出錯原因:numpy的版本出現問題,導致不能與PyTorch的版本進行匹配,具體為numpy版本過低,
解決方案:升級numpy的版本,激活環境后輸入下面的命令:
conda upgrade numpy
End,
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/226874.html
標籤:python
上一篇:Windows下用Anaconda3安裝PyTorch環境(詳細教程)
下一篇:flutter環境部署
