如何搭建pytorch環(huán)境的方法步驟
1.conda創(chuàng)建虛擬環(huán)境pytorch_gpu
conda create -n pytorch_gpu python=3.6
創(chuàng)建虛擬環(huán)境還是相對(duì)較快的，它會(huì)自動(dòng)為本環(huán)境安裝一些基本的庫(kù)，等待時(shí)間無(wú)需很長(zhǎng)，成功之后界面如下所示：
2.切換到pytorch環(huán)境
使用如下命令，切換到我們剛剛創(chuàng)建好的pytorch虛擬環(huán)境，這樣我們避免與其它python環(huán)境之間的干擾。
conda activeta pytorch_gpu
切換成功之后就會(huì)看到在路徑前邊顯示我們已經(jīng)進(jìn)入該虛擬環(huán)境。
3.安裝幾個(gè)常用庫(kù)（也可暫時(shí)不安）
conda install pandas jupyter notebook
4.安裝pytorch
4.1進(jìn)入官網(wǎng)查看要下載的版本
查看對(duì)應(yīng)的版本，這里是官方鏈接：
4.2 根據(jù)系 統(tǒng)信息及cuda版本選擇對(duì)應(yīng)toolkit
這里最主要的是那個(gè)CUDA的版本，此處我選擇的是10.1，是因?yàn)槲业碾娔X的 cuda版本信息就是這樣的。具體的查看方法可在4.5節(jié)查閱。
4.3復(fù)制上圖中最后一行代碼到pytorch環(huán)境終端
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
這里是下載過(guò)程截圖：
但是在下載過(guò)程中torchvision以及pytorch沒(méi)有下載成功，
因?yàn)檫@是去其官網(wǎng)下載，下載速度很慢，在上圖中我們也可以看出是因?yàn)榫W(wǎng)絡(luò)錯(cuò)誤,網(wǎng)絡(luò)上也有幾種其他的方法，此處我沒(méi)去驗(yàn)證，我還是讓電腦重新下載的，等待時(shí)間挺長(zhǎng)的，但是因?yàn)槭窃谕砩舷螺d的，一早起來(lái)就好了
4.4 驗(yàn)證pytorch是否安裝成功
此時(shí)直接輸入 python，即可成功進(jìn)入：
而后輸入如下指令，查看torch是否安裝成功
>>> import torch >>> x=torch.randn(4,4) >>> print(x)
正常情況下是出現(xiàn)這個(gè)界面的：
驗(yàn)證完成之后，可以quit()保存退出。
4.5 如何查看自己電腦cuda版本
4.5.1 windows如何查看
NVDIA控制面板–>幫助–>系統(tǒng)信息
組件–>NVCUDA.DLL 可以查看CUDA版本
這里我的顯示是10.1,所以我上邊下載的版本也是10.1的，這里的版本要對(duì)應(yīng)上，否則會(huì)出現(xiàn)問(wèn)題。
4.5.2 linux如何查看
打開終端，輸入：nvcc -V
nvcc -V
或者如下方式查看：
CUDA:
cat /usr/local/cuda/version.txt
cudnn:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
到此這篇關(guān)于如何搭建pytorch環(huán)境的方法步驟的文章就介紹到這了,更多相關(guān)pytorch搭建環(huán)境內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了python查找與排序算法詳解(示圖+代碼)，文章通過(guò)二分查找展開主題詳細(xì)內(nèi)容，需要的朋友可以參考一下2022-07-07
python使用多線程備份數(shù)據(jù)庫(kù)的步驟
在日常服務(wù)器運(yùn)維工作中，備份數(shù)據(jù)庫(kù)是必不可少的，剛工作那會(huì)看到公司都是用shell腳本循環(huán)備份數(shù)據(jù)庫(kù)，到現(xiàn)在自己學(xué)習(xí)python語(yǔ)言后，利用多進(jìn)程多線程相關(guān)技術(shù)來(lái)實(shí)現(xiàn)并行備份數(shù)據(jù)庫(kù)，充分利用服務(wù)器資源，提高備份速度。2021-05-05
Python獲取接口請(qǐng)求耗時(shí)的方法詳解
你想知道我們請(qǐng)求一個(gè)url的時(shí)候，握手和請(qǐng)求資源分別占用多長(zhǎng)時(shí)間么？今天我們就來(lái)使用python寫個(gè)小案例來(lái)看看，感興趣的可以跟隨小編一起了解一下2023-04-04
這篇文章主要介紹了Python判斷兩個(gè)對(duì)象相等的原理,需要的朋友可以參考下2017-12-12
詳解python常用命令行選項(xiàng)與環(huán)境變量
這篇文章主要介紹了python常用命令行選項(xiàng)與環(huán)境變量，本文給大家介紹的非常詳細(xì)，具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-02-02
今天小編就為大家分享一篇PyQt5 在label顯示的圖片中繪制矩形的方法，具有很好的參考價(jià)值。希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-06-06
python引用(import)某個(gè)模塊提示沒(méi)找到對(duì)應(yīng)模塊的解決方法
今天小編就為大家分享一篇python引用(import)某個(gè)模塊提示沒(méi)找到對(duì)應(yīng)模塊的解決方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-01-01
Matplotlib是Python的繪圖庫(kù)，它能讓使用者很輕松地將數(shù)據(jù)圖形化，并且提供多樣化的輸出格式。本文將為大家介紹如何用matplotlib繪制灰度和彩色直方圖，感興趣的朋友可以學(xué)習(xí)一下2021-12-12
教你如何用python操作攝像頭以及對(duì)視頻流的處理
這篇文章主要介紹了教你如何用python操作攝像頭以及對(duì)視頻流的處理，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-10-10
在本篇文章里小編給大家整理的是關(guān)于python腳本和網(wǎng)頁(yè)的區(qū)別點(diǎn)總結(jié)，有興趣的朋友們可以學(xué)習(xí)下。2020-07-07