win10系統(tǒng)配置GPU版本Pytorch的詳細教程

一、安裝cuda

1、在英偉達官網(wǎng)下載最新版的cuda驅(qū)動

https://developer.nvidia.com/zh-cn/cuda-downloads

都選上就行了，然后一路默認安裝

輸入nvcc -V查看是否安裝成功

二、安裝pycuda

1、在控制臺中輸入pip install pycuda 安裝pycuda
2、在環(huán)境變量中添加cl.exe

3、測試pycuda是否正常運行

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計算結(jié)果跟CPU一樣

三、安裝GPU版本的pytorch

進入pytorch官網(wǎng)

附：遇到的問題及解決辦法：

1、剛裝好的時候可以加載pycuda，但是無法找到模塊

經(jīng)過我的分析應(yīng)該是剛安裝好系統(tǒng)不是很穩(wěn)定，我重啟了一下電腦解決了問題

2、更改了PATH變量后無反映

修改系統(tǒng)變量后要重啟電腦后才能生效

3、在系統(tǒng)變量中添加某個程序時，要添加他上一級目錄，不用指定到具體哪個程序。

到此這篇關(guān)于win10系統(tǒng)配置GPU版本Pytorch的文章就介紹到這了,更多相關(guān)win10 GPU配置Pytorch內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！