前言

以下述環(huán)境為例：
python=3.7
pytorch=1.7.1
torchvision==0.8.2
cudatoolkit=10.1

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

一、anaconda安裝

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

二、pytorch安裝

1.新建環(huán)境

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

conda create -n pytorch python=3.7 -y

#創(chuàng)建一個名為pytorch的環(huán)境，python版本3.7，確認

2.激活環(huán)境

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系統(tǒng)Pytorch環(huán)境搭建過程的文章就介紹到這了,更多相關Win10搭建Pytorch環(huán)境內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！