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

進(jìn)入pytorch官網(wǎng)

附：遇到的問(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)文章希望大家以后多多支持腳本之家！