PyTorch 安裝指南

PyTorch 安裝指南

本博客主要介紹在 Anaconda虛擬環境中建立 PyTorch環境，默認已經安裝成功 Anaconda軟件。

Anaconda Individual Edition的官方下載連接:

https://www.anaconda.com/products/individual

按照本身的操做系統，選擇須要的版本號，隨後傻瓜式安裝便可。具體的安裝及使用能夠參考如下連接：

http://www.javashuo.com/article/p-kfjrqgcr-dt.html

建立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版本號，這裏能夠參考連接

http://www.javashuo.com/article/p-xalydbix-nq.html

我本身的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。