Win10系統(tǒng)下Pytorch環(huán)境的搭建過程

前言

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

pytorch和cuda版本對應(yīng)參考下圖：

一、anaconda安裝

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

二、pytorch安裝

1.新建環(huán)境

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

conda create -n pytorch python=3.7 -y

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

2.激活環(huán)境

activate pytorch

3.安裝pytorch

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

#該命令可以到官網(wǎng)https://pytorch.org/get-started/locally/查看，選擇對應(yīng)選項，然后將run this command復(fù)制到命令行中。

總結(jié)

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

到此這篇關(guān)于Win10系統(tǒng)Pytorch環(huán)境搭建過程的文章就介紹到這了,更多相關(guān)Win10搭建Pytorch環(huán)境內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！