前言

以下述環境為例：
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環境內容請搜索腳本之家以前的文章或繼續瀏覽下面的相關文章希望大家以后多多支持腳本之家！