win10系統(tǒng)配置GPU版本Pytorch的詳細(xì)教程
一、安裝cuda
1、在英偉達(dá)官網(wǎng)下載最新版的cuda驅(qū)動
https://developer.nvidia.com/zh-cn/cuda-downloads
都選上就行了，然后一路默認(rèn)安裝
輸入nvcc -V查看是否安裝成功
二、安裝pycuda
1、在控制臺中輸入pip install pycuda 安裝pycuda
2、在環(huán)境變量中添加cl.exe
3、測試pycuda是否正常運(yùn)行
import pycuda.driver as drv
import pycuda.tools,pycuda.autoinit,numpy
import numpy.linalg as la
from pycuda.compiler import SourceModule
mod = SourceModule("""
__global__ void multiply_them(float *dest, float *a, float *b)
{
const int i = threadIdx.x;
dest[i] = a[i] * b[i];
}
""")
multiply_them = mod.get_function("multiply_them")
a = numpy.random.randn(400).astype(numpy.float32)
b = numpy.random.randn(400).astype(numpy.float32)
dest = numpy.zeros_like(a)
multiply_them(drv.Out(dest), drv.In(a), drv.In(b),block=(400,1,1))
print(dest-a*b)
全為0說明GPU計(jì)算結(jié)果跟CPU一樣
三、安裝GPU版本的pytorch
附：遇到的問題及解決辦法：
1、剛裝好的時(shí)候可以加載pycuda，但是無法找到模塊
經(jīng)過我的分析應(yīng)該是剛安裝好系統(tǒng)不是很穩(wěn)定，我重啟了一下電腦解決了問題
2、更改了PATH變量后無反映
修改系統(tǒng)變量后要重啟電腦后才能生效
3、在系統(tǒng)變量中添加某個(gè)程序時(shí)，要添加他上一級目錄，不用指定到具體哪個(gè)程序。
到此這篇關(guān)于win10系統(tǒng)配置GPU版本Pytorch的文章就介紹到這了,更多相關(guān)win10 GPU配置Pytorch內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
本文主要介紹了python中numpy矩陣的零填充的示例代碼，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2022-08-08
在python里，可能大家已從運(yùn)算表里找到了，即 （“+”），但是與PHP不同的是兩個(gè)對象相加，兩個(gè)對象必須是同類型的,不同類型的兩個(gè)對象是不能相加的.要實(shí)現(xiàn)像上面PHP代碼里的運(yùn)算，必須保證兩對象類型一致。2014-09-09
python進(jìn)階學(xué)習(xí)實(shí)時(shí)目標(biāo)跟蹤示例詳解
這篇文章主要為大家介紹了python進(jìn)階學(xué)習(xí)實(shí)時(shí)目標(biāo)跟蹤示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-03-03
這篇文章主要介紹了簡單了解python反射機(jī)制的一些知識，反射機(jī)制就是在運(yùn)行時(shí)，動態(tài)的確定對象的類型，并可以通過字符串調(diào)用對象屬性、方法、導(dǎo)入模塊，是一種基于字符串的事件驅(qū)動。,需要的朋友可以參考下2019-07-07
最近在搗鼓配音視頻課件的制作方法，發(fā)現(xiàn)使用Moviepy進(jìn)行合成比圖形操作界面的合成軟件效果更好，可以完美的解決音頻和ppt材料的協(xié)同問題，下面就詳細(xì)介紹一下這個(gè)過程，供ppt視頻課件制作生手提供一個(gè)可以高效制作視頻的方法2021-06-06
這篇文章主要為大家詳細(xì)介紹了python處理大日志文件的的相關(guān)方法，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2019-07-07
深入解析Python中BeautifulSoup4的基礎(chǔ)知識與實(shí)戰(zhàn)應(yīng)用
BeautifulSoup4正是一款功能強(qiáng)大的解析器,能夠輕松解析HTML和XML文檔,本文將介紹BeautifulSoup4的基礎(chǔ)知識,并通過實(shí)際代碼示例進(jìn)行演示,感興趣的可以了解下2024-02-02
Python實(shí)現(xiàn)批量壓縮文件/文件夾zipfile的使用
本文主要介紹了Python實(shí)現(xiàn)批量壓縮文件/文件夾zipfile的使用，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2021-09-09
Python ORM框架SQLAlchemy學(xué)習(xí)筆記之安裝和簡單查詢實(shí)例
這篇文章主要介紹了Python ORM框架SQLAlchemy學(xué)習(xí)筆記之安裝和簡單查詢實(shí)例,簡明入門教程,需要的朋友可以參考下2014-06-06
在本篇文章里小編給大家整理的是一篇關(guān)于python分?jǐn)?shù)實(shí)例用法的相關(guān)內(nèi)容，有興趣的朋友們可以學(xué)習(xí)下。2021-04-04