如何安裝PyTorch

這篇文章主要介紹了如何安裝PyTorch，具有一定借鑒價值，需要的朋友可以參考下。希望大家閱讀完這篇文章后大有收獲。下面讓小編帶著大家一起了解一下。

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

Anaconda Inpidual Edition的官方下載鏈接:

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

按照自己的操作系統，選擇需要的版本號，隨后傻瓜式安裝即可。具體的安裝及使用可以參考以下鏈接：

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。

PyTorch 安裝指南

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

Anaconda Inpidual Edition的官方下載鏈接:

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

按照自己的操作系統，選擇需要的版本號，隨后傻瓜式安裝即可。具體的安裝及使用可以參考以下鏈接：

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

感謝你能夠認真閱讀完這篇文章，希望小編分享如何安裝PyTorch內容對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業資訊頻道，遇到問題就找億速云，詳細的解決方法等著你來學習!