Win10系統下Pytorch環境的搭建過程

前言

以下述環境為例：
python=3.7
pytorch=1.7.1
torchvision==0.8.2
cudatoolkit=10.1

pytorch和cuda版本對應參考下圖：

一、anaconda安裝

做機器學習和深度學習離不開環境和包管理，要養成每個工程單獨設置環境的習慣！！！Anaconda和Miniconda兩個自選（推薦前者，後者可以用在筆記本上節省空間）。
anaconda
鏈接: link.
miniconda
鏈接: link.

二、pytorch安裝

1.新建環境

新版Anaconda現在默認python為最新版本，所以需要哪個版本得自己設置，常用新建環境命令在命令行中為：

conda create -n pytorch python=3.7 -y

#創建一個名為pytorch的環境，python版本3.7，確認

2.激活環境

activate pytorch

3.安裝pytorch

conda install pytorch==1.7.1 torchvision==0.8.2 cudatoolkit=10.1 -c pytorch

#該命令可以到官網https://pytorch.org/get-started/locally/查看，選擇對應選項，然後將run this command復制到命令行中。

總結

步驟很簡單，但是過程中難免遇到各種問題，尤其是下載中進度不動或者中斷瞭，重復操作即可，換源也是一種辦法。

到此這篇關於Win10系統Pytorch環境搭建過程的文章就介紹到這瞭,更多相關Win10搭建Pytorch環境內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet！

推薦閱讀：

• anaconda安裝pytorch1.7.1和torchvision0.8.2的方法(親測可用)
• 關於PyTorch環境配置及安裝教程(Windows10)
• 解決安裝pytorch因網速問題失敗的情況
• 解決anaconda安裝pytorch報錯找不到包的問題
• Python服務器創建虛擬環境跑代碼