win10系統(tǒng)配置GPU版本Pytorch的詳細(xì)教程
一、安裝cuda
1、在英偉達(dá)官網(wǎng)下載最新版的cuda驅(qū)動(dòng)
https://developer.nvidia.com/zh-cn/cuda-downloads
都選上就行了，然后一路默認(rèn)安裝
輸入nvcc -V查看是否安裝成功
二、安裝pycuda
1、在控制臺(tái)中輸入pip install pycuda 安裝pycuda
2、在環(huán)境變量中添加cl.exe
3、測(cè)試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說(shuō)明GPU計(jì)算結(jié)果跟CPU一樣
三、安裝GPU版本的pytorch
附：遇到的問(wèn)題及解決辦法：
1、剛裝好的時(shí)候可以加載pycuda，但是無(wú)法找到模塊
經(jīng)過(guò)我的分析應(yīng)該是剛安裝好系統(tǒng)不是很穩(wěn)定，我重啟了一下電腦解決了問(wèn)題
2、更改了PATH變量后無(wú)反映
修改系統(tǒng)變量后要重啟電腦后才能生效
3、在系統(tǒng)變量中添加某個(gè)程序時(shí)，要添加他上一級(jí)目錄，不用指定到具體哪個(gè)程序。
到此這篇關(guān)于win10系統(tǒng)配置GPU版本Pytorch的文章就介紹到這了,更多相關(guān)win10 GPU配置Pytorch內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
Python中用于計(jì)算對(duì)數(shù)的log()方法
這篇文章主要介紹了Python中用于計(jì)算對(duì)數(shù)的log()方法,是Python入門(mén)基礎(chǔ)中的必會(huì)的方法,需要的朋友可以參考下2015-05-05
這篇文章主要介紹了Python教程按照字典的鍵或值進(jìn)行排序方法解析，有需要的朋友可以借鑒參考下，希望可以有所幫助，祝大家多多進(jìn)步，早日升職加薪2021-09-09
python爬蟲(chóng)之代理ip正確使用方法實(shí)例
在爬蟲(chóng)的過(guò)程中,我們經(jīng)常會(huì)遇見(jiàn)很多網(wǎng)站采取了防爬蟲(chóng)技術(shù),或者說(shuō)因?yàn)樽约翰杉W(wǎng)站信息的強(qiáng)度和采集速度太大,給對(duì)方服務(wù)器帶去了太多的壓力,下面這篇文章主要給大家介紹了關(guān)于python爬蟲(chóng)之代理ip正確使用方法的相關(guān)資料,需要的朋友可以參考下2022-07-07
python paramiko模塊學(xué)習(xí)分享
這篇文章主要為大家分享了python paramiko模塊的學(xué)習(xí)資料，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2017-08-08
Python保留指定位數(shù)小數(shù)的5種方法總結(jié)
很多小伙伴在學(xué)習(xí)python的時(shí)候可能會(huì)遇到對(duì)數(shù)據(jù)進(jìn)行格式化輸出的需求,其中最常見(jiàn)的需求為保留幾位小數(shù),這篇文章主要給大家介紹了關(guān)于Python保留指定位數(shù)小數(shù)的5種方法,需要的朋友可以參考下2023-08-08
Python在畫(huà)圖時(shí)使用特殊符號(hào)的方法總結(jié)
在制作圖表時(shí)，如果遇到需要利用特殊符號(hào)進(jìn)行表示時(shí)該怎么辦呢？不用慌，這篇文章為大家總結(jié)了python畫(huà)圖中使用各種特殊符號(hào)的方式，需要的可以參考一下2022-04-04
python網(wǎng)絡(luò)編程之進(jìn)程詳解
這篇文章主要為大家介紹了python網(wǎng)絡(luò)編程之進(jìn)程，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來(lái)幫助2022-01-01
這篇文章主要介紹了python tkinter 獲得按鈕的文本值，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04
python mysql項(xiàng)目實(shí)戰(zhàn)及框架搭建過(guò)程
本文給大家分享python mysql項(xiàng)目實(shí)戰(zhàn)框架搭建過(guò)程，通過(guò)實(shí)例代碼給大家講解python mysql項(xiàng)目實(shí)戰(zhàn)的相關(guān)知識(shí)，需要的朋友參考下吧2021-06-06
Python實(shí)現(xiàn)從log日志中提取ip的方法【正則提取】
這篇文章主要介紹了Python實(shí)現(xiàn)從log日志中提取ip的方法,涉及Python文件讀取、數(shù)據(jù)遍歷、正則匹配等相關(guān)操作技巧,需要的朋友可以參考下2018-03-03