脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → win10 GPU配置Pytorch

win10系统配置GPU版本Pytorch的详细教程

 更新时间：2021年04月29日 08:55:08   作者：SGchi  
这篇文章主要介绍了win10系统配置GPU版本Pytorch,本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下

一、安装cuda

1、在英伟达官网下载最新版的cuda驱动

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

在这里插入图片描述

都选上就行了，然后一路默认安装

输入nvcc -V查看是否安装成功

在这里插入图片描述

二、安装pycuda

1、在控制台中输入pip install pycuda 安装pycuda
2、在环境变量中添加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计算结果跟CPU一样

三、安装GPU版本的pytorch

进入pytorch官网

在这里插入图片描述

附：遇到的问题及解决办法：

1、刚装好的时候可以加载pycuda，但是无法找到模块

在这里插入图片描述

在这里插入图片描述

经过我的分析应该是刚安装好系统不是很稳定，我重启了一下电脑解决了问题

2、更改了PATH变量后无反映

修改系统变量后要重启电脑后才能生效

3、在系统变量中添加某个程序时，要添加他上一级目录，不用指定到具体哪个程序。

到此这篇关于win10系统配置GPU版本Pytorch的文章就介绍到这了,更多相关win10 GPU配置Pytorch内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

相关文章

  • Python中用于计算对数的log()方法

    Python中用于计算对数的log()方法

    这篇文章主要介绍了Python中用于计算对数的log()方法,是Python入门基础中的必会的方法,需要的朋友可以参考下
    2015-05-05
  • Python教程按照字典的键或值进行排序方法解析

    Python教程按照字典的键或值进行排序方法解析

    这篇文章主要介绍了Python教程按照字典的键或值进行排序方法解析，有需要的朋友可以借鉴参考下，希望可以有所帮助，祝大家多多进步，早日升职加薪
    2021-09-09
  • python爬虫之代理ip正确使用方法实例

    python爬虫之代理ip正确使用方法实例

    在爬虫的过程中,我们经常会遇见很多网站采取了防爬虫技术,或者说因为自己采集网站信息的强度和采集速度太大,给对方服务器带去了太多的压力,下面这篇文章主要给大家介绍了关于python爬虫之代理ip正确使用方法的相关资料,需要的朋友可以参考下
    2022-07-07
  • python paramiko模块学习分享

    python paramiko模块学习分享

    这篇文章主要为大家分享了python paramiko模块的学习资料，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2017-08-08
  • Python保留指定位数小数的5种方法总结

    Python保留指定位数小数的5种方法总结

    很多小伙伴在学习python的时候可能会遇到对数据进行格式化输出的需求,其中最常见的需求为保留几位小数,这篇文章主要给大家介绍了关于Python保留指定位数小数的5种方法,需要的朋友可以参考下
    2023-08-08
  • Python在画图时使用特殊符号的方法总结

    Python在画图时使用特殊符号的方法总结

    在制作图表时，如果遇到需要利用特殊符号进行表示时该怎么办呢？不用慌，这篇文章为大家总结了python画图中使用各种特殊符号的方式，需要的可以参考一下
    2022-04-04
  • python网络编程之进程详解

    python网络编程之进程详解

    这篇文章主要为大家介绍了python网络编程之进程，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助
    2022-01-01
  • python tkinter 获得按钮的文本值

    python tkinter 获得按钮的文本值

    这篇文章主要介绍了python tkinter 获得按钮的文本值，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2021-04-04
  • python mysql项目实战及框架搭建过程

    python mysql项目实战及框架搭建过程

    本文给大家分享python mysql项目实战框架搭建过程，通过实例代码给大家讲解python mysql项目实战的相关知识，需要的朋友参考下吧
    2021-06-06
  • Python实现从log日志中提取ip的方法【正则提取】

    Python实现从log日志中提取ip的方法【正则提取】

    这篇文章主要介绍了Python实现从log日志中提取ip的方法,涉及Python文件读取、数据遍历、正则匹配等相关操作技巧,需要的朋友可以参考下
    2018-03-03

最新评论