win10系统配置GPU版本Pytorch的详细教程
作者：SGchi 发布时间：2023-07-21 19:30:38
标签：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、在系统变量中添加某个程序时，要添加他上一级目录，不用指定到具体哪个程序。
来源：https://blog.csdn.net/SGchi/article/details/116241998
0
投稿猜你喜欢
在实际的项目中，我们一般都会建立三个环境：开发、测试和生产环境，这三种环境会使用不同的配置组合，为了能方便地切换配置，我们可以为不同的环境创- 问题：关于如何生成随机记录（二）如何从指定表中随机抽取一定量的记录？sql server 中 select top 10 * fr
- Django 分页功能的实现，供大家参考，具体内容如下创建项目创建APP，添加APP这些就不在多说我们这次重点来看到视图函数下面是路由设置视
- 什么是品牌的视觉传达品牌，这个熟悉而又陌生的名词，有时总会让人产生误解。品牌很广，广到一个意会颇深的战略发展理念；品牌很小，小到一个清晰可见
- 本文实例讲述了Python常用字符串替换函数strip、replace及sub用法。分享给大家供大家参考，具体如下：今天在做一道今年秋季招聘
- 修改MySQL下的默认mysql数据库的user表，删除所有host为localhost记录，另外添加一些其他记录，重新启动MySQL服务器
- reflow是个神奇的东西，之前Realazy说到过这个reflow，我摘出其中的重点：在CSS规范中有一个渲染对象的概念，通常用一个盒子（
- 前言pydub是Python中用户处理音频文件的一个库。本文主要介绍了关于Python音频处理库pydub使用的相关内容，分享出来供大家参考
- 前几天有个人退群了。起因很简单，他问了一个问题，没人回答，于是说要退群，后来我看到了，给了个链接，说这个问题已经说过好多遍了，于是他就退了。
- 这个是JS控制图片滚动的效果，当鼠标结果新闻标题时开始滚动到对应的图片，可以作为图片新闻。效果图：<!DOCTYPE HTML PUB
- CSS 和 JavaScript 的压缩已经很成熟，各大网站都在使用。HTML 的压缩（特指去除空白字符和注释），除了 Google 等搜索
- 发现问题今天在使用ImageTk.photoImage()显示图片时，当把包含该函数放在自定义函数里时，不能正常显示，移到函数为又可正常显示
- 人一旦习惯了某些东西就很难去改，以及各种各样的原因，新的浏览器越来越多，而老的总淘汰不了。增长总是快于消亡导致了浏览器兼容是成了谈不完的话题
- 注：本文是应Alan邀请为《CSS布局实录》写的一个web标准入门指导。书已经上市近一年了，现在摘选出来，给初学者一个参考。希望了解更多实现
- <?php/** * 发送文件 * * @author: legend(legendsky@hotmai
- EM算法实例通过实例可以快速了解EM算法的基本思想，具体推导请点文末链接。图a是让我们预热的，图b是EM算法的实例。这是一个抛硬币的例子，H
- “Be conservative in what you send; be liberal in what you accept. &nbs
- 安装pillow(python的图形界面库)第一种方法在Dos界面输入pip install pillow(但是不知为何总是失败)；搞了好几
- 函数声明为：func Notify(c chan<- os.Signal, sig ...os.Signal)官方描述：Notify函
- 数据库中有一字段type_code，有中文类型和中文类型编码，现在对type_code字段的数据进行统计处理，编码对应的字典如下：{'
