如何搭建pytorch環(huán)境的方法步驟
1.conda創(chuàng)建虛擬環(huán)境pytorch_gpu
conda create -n pytorch_gpu python=3.6
創(chuàng)建虛擬環(huán)境還是相對較快的，它會自動為本環(huán)境安裝一些基本的庫，等待時間無需很長，成功之后界面如下所示：
2.切換到pytorch環(huán)境
使用如下命令，切換到我們剛剛創(chuàng)建好的pytorch虛擬環(huán)境，這樣我們避免與其它python環(huán)境之間的干擾。
conda activeta pytorch_gpu
切換成功之后就會看到在路徑前邊顯示我們已經(jīng)進(jìn)入該虛擬環(huán)境。
3.安裝幾個常用庫（也可暫時不安）
conda install pandas jupyter notebook
4.安裝pytorch
4.1進(jìn)入官網(wǎng)查看要下載的版本
查看對應(yīng)的版本，這里是官方鏈接：
4.2 根據(jù)系 統(tǒng)信息及cuda版本選擇對應(yīng)toolkit
這里最主要的是那個CUDA的版本，此處我選擇的是10.1，是因為我的電腦的 cuda版本信息就是這樣的。具體的查看方法可在4.5節(jié)查閱。
4.3復(fù)制上圖中最后一行代碼到pytorch環(huán)境終端
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
這里是下載過程截圖：
但是在下載過程中torchvision以及pytorch沒有下載成功，
因為這是去其官網(wǎng)下載，下載速度很慢，在上圖中我們也可以看出是因為網(wǎng)絡(luò)錯誤,網(wǎng)絡(luò)上也有幾種其他的方法，此處我沒去驗證，我還是讓電腦重新下載的，等待時間挺長的，但是因為是在晚上下載的，一早起來就好了
4.4 驗證pytorch是否安裝成功
此時直接輸入 python，即可成功進(jìn)入：
而后輸入如下指令，查看torch是否安裝成功
>>> import torch >>> x=torch.randn(4,4) >>> print(x)
正常情況下是出現(xiàn)這個界面的：
驗證完成之后，可以quit()保存退出。
4.5 如何查看自己電腦cuda版本
4.5.1 windows如何查看
NVDIA控制面板–>幫助–>系統(tǒng)信息
組件–>NVCUDA.DLL 可以查看CUDA版本
這里我的顯示是10.1,所以我上邊下載的版本也是10.1的，這里的版本要對應(yīng)上，否則會出現(xià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)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要為大家詳細(xì)介紹了如何使用Python和PyQt5開發(fā)一個截圖工具,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2024-12-12
這篇文章主要介紹了Python Scrapy圖片爬取原理及代碼實例,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-06-06
這篇文章主要介紹了Scrapy爬蟲文件批量運行的實現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
這篇文章主要介紹了Python 列表的索引取值,本節(jié)重點掌握多次索引取值的語法：列表[索引][索引]，結(jié)合示例代碼給大家介紹的非常詳細(xì)，需要的朋友可以參考下2022-09-09
python中csv文件創(chuàng)建、讀取及修改等操作實例
很多程序在處理數(shù)據(jù)時都會碰到csv這種格式的文件,下面這篇文章主要給大家介紹了關(guān)于python中csv文件創(chuàng)建、讀取及修改等操作的相關(guān)資料,文中通過實例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-05-05
Flask sqlalchemy一對多與多對一與一對一及多對多關(guān)系介紹
這篇文章主要介紹了Flask sqlalchemy一對多與多對一與一對一及多對多關(guān)系介紹，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-09-09
這篇文章主要介紹了Python讀取GSMap數(shù)據(jù)的問題,本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-03-03
python打包exe文件并隱藏執(zhí)行CMD命令窗口問題
這篇文章主要介紹了python打包exe文件并隱藏執(zhí)行CMD命令窗口問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-01-01
python經(jīng)典百題之static定義靜態(tài)變量的三種方法
日常腳本編寫過程中時常會用到python的靜態(tài)方法、實例方法、類方法,下面這篇文章主要給大家介紹了關(guān)于python經(jīng)典百題之static定義靜態(tài)變量的三種方法,需要的朋友可以參考下2024-09-09
詳解常用查找數(shù)據(jù)結(jié)構(gòu)及算法（Python實現(xiàn)）
本篇文章主要介紹了Python實現(xiàn)常用查找數(shù)據(jù)結(jié)構(gòu)及算法，具有一定的參考價值，有興趣的可以了解一下。2016-12-12