快捷導(dǎo)航

PyTorch 檢查GPU版本是否安裝成功的操作

更新時(shí)間：2021年03月17日 08:50:31 作者：敲代碼的小風(fēng)

這篇文章主要介紹了PyTorch 檢查GPU版本是否安裝成功的操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

anaconda命令行下檢查:

(base) PS C:\Users\chenxuqi> conda deactivate
PS C:\Users\chenxuqi> conda activate ssd
(ssd) PS C:\Users\chenxuqi> python
Python 3.6.12 |Anaconda, Inc.| (default, Sep 9 2020, 00:29:25) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
0.4.0
>>> print(torch.cuda.is_available())
True
>>>
>>>
>>>

cmd命令行下檢查cuda安裝:

Microsoft Windows [版本 10.0.18363.1139]
(c) 2019 Microsoft Corporation。保留所有權(quán)利。

C:\Users\chenxuqi>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:32_Central_Daylight_Time_2017
Cuda compilation tools, release 9.0, V9.0.176

C:\Users\chenxuqi>
C:\Users\chenxuqi>

執(zhí)行使用GPU的代碼:

import time
import torch 

##################################################

for i in range(1,10):
  start = time.time()
  a = torch.FloatTensor(i*100,1000,1000)
  a = a.cuda() #a = a
  a = torch.matmul(a,a)
  end = time.time() - start
  print(end)

執(zhí)行結(jié)果:

注意，這里顯存太小，溢出了...但是安裝是成功的...

Windows PowerShell

嘗試新的跨平臺(tái) PowerShell https://aka.ms/pscore6

PS C:\Users\chenxuqi\Desktop\新建文件夾> & 'D:\Anaconda3\envs\ssd\python.exe' 'c:\Users\chenxuqi\.vscode\extensions\ms-python.python-2020.10.332292344\pythonFiles\lib\python\debugpy\launcher' '50571' '--' 'c:\Users\chenxuqi\Desktop\新建文件夾\testGPU.py'
3.6260359287261963
0.6305170059204102
0.9055967330932617
1.3199987411499023
1.5979139804840088
2.0483360290527344
THCudaCheck FAIL file=c:\programdata\miniconda3\conda-bld\pytorch_1524549877902\work\aten\src\thc\generic/THCStorage.cu line=58 error=2 : out of memory
Traceback (most recent call last):
 File "c:\Users\chenxuqi\Desktop\新建文件夾\testGPU.py", line 10, in <module>
  a = torch.matmul(a,a)
RuntimeError: cuda runtime error (2) : out of memory at c:\programdata\miniconda3\conda-bld\pytorch_1524549877902\work\aten\src\thc\generic/THCStorage.cu:58
PS C:\Users\chenxuqi\Desktop\新建文件夾> conda activate ssd
PS C:\Users\chenxuqi\Desktop\新建文件夾>

補(bǔ)充：pytorch離線安裝，驗(yàn)證gpu版安裝成功

使用conda 命令在線安裝pytorch會(huì)下載中斷，添加pip清華大學(xué)鏡像源https://pypi.tuna.tsinghua.edu.cn/simple/中下好torch,安裝包，然后使用pip命令

pip install "下在的安裝包的路徑"

例如：

pip install "C:\Users\28614\Desktop\pytorch-nightly-cpu-1.0.0.dev20181222-py3.7_cpu_0.tar.bz2"

安裝gpu版，需要安裝cuda,和cudnn。

驗(yàn)證gpu版是否安裝成功

import torch
print(torch.cuda.is_available())

返回True，則安裝成功

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章:

相關(guān)文章

Pandas修改DataFrame列名的兩種方法實(shí)例
這篇文章主要給大家介紹了關(guān)于Pandas修改DataFrame列名的兩種方法,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),對(duì)大家學(xué)習(xí)或者使用Pandas具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2022-03-03
Django如何判斷訪問(wèn)來(lái)源是PC端還是手機(jī)端
這篇文章主要介紹了Django如何判斷訪問(wèn)來(lái)源是PC端還是手機(jī)端問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
2023-05-05
Python數(shù)據(jù)庫(kù)格式化輸出文檔的思路與方法
這篇文章主要給大家介紹了關(guān)于Python數(shù)據(jù)庫(kù)格式化輸出文檔的思路與方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧
2021-03-03
Python中操作文件之write()方法的使用教程
這篇文章主要介紹了Python中操作文件之write()方法的使用教程,是Python入門(mén)學(xué)習(xí)中的基礎(chǔ)知識(shí),需要的朋友可以參考下
2015-05-05
python非對(duì)稱加密算法RSA實(shí)現(xiàn)原理與應(yīng)用詳解
RSA加密算法是一種非對(duì)稱加密算法,RSA算法的安全性基于大數(shù)分解的困難性,即已知兩個(gè)大素?cái)?shù)p和q的乘積n,求解p和q非常困難,RSA算法廣泛應(yīng)用于數(shù)據(jù)加密和數(shù)字簽名等領(lǐng)域,本文將詳細(xì)介紹如何在Python中使用RSA算法進(jìn)行加密和解密,需要的朋友可以參考下
2024-09-09
python 爬取豆瓣網(wǎng)頁(yè)的示例
這篇文章主要介紹了python 爬取豆瓣網(wǎng)頁(yè)的示例，幫助大家更好的理解和學(xué)習(xí)使用python，感興趣的朋友可以了解下
2021-04-04
解讀requests.session()獲取Cookies全過(guò)程
這篇文章主要介紹了解讀requests.session()獲取Cookies全過(guò)程，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
2023-02-02
Pytorch卷積神經(jīng)網(wǎng)絡(luò)遷移學(xué)習(xí)的目標(biāo)及好處
這篇文章主要為大家介紹了Pytorch卷積神經(jīng)網(wǎng)絡(luò)遷移學(xué)習(xí)的目標(biāo)實(shí)現(xiàn)代碼及好處介紹，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪
2022-05-05
Django開(kāi)發(fā)的簡(jiǎn)易留言板案例詳解
這篇文章主要介紹了Django開(kāi)發(fā)的簡(jiǎn)易留言板,結(jié)合實(shí)例形式詳細(xì)分析了基于Python框架Django開(kāi)發(fā)留言板的具體文件結(jié)構(gòu)、流程步驟與相關(guān)操作技巧,需要的朋友可以參考下
2018-12-12
使用Python?http.server模塊共享文件的方法詳解
大家好,今天給大家介紹一下Python標(biāo)準(zhǔn)庫(kù)中的http.server模塊,這個(gè)模塊提供了一種簡(jiǎn)單的方式來(lái)快速啟動(dòng)一個(gè)HTTP服務(wù)器,文中給大家介紹了使用Python?http.server模塊共享文件的方法,需要的朋友可以參考下
2024-05-05