Windows的torch + Cuda + cuDNN_cuda必须安装在c盘吗-程序员宅基地

技术标签: python  深度学习  windows  

1、CUDA下载

使用NVIDIA-SMI命令查看合适自己电脑的CUDA版本

nvidia-smi

进入NVIDIA官网下载CUDA

https://developer.nvidia.com/cuda-toolkit-archive

找到合适电脑版本的CUDA版本

点击CUDA Toolkit 10.2进入下载页面,选择操作系统版本,架构,电脑系统版本,安装类型选择local,然后选择base installer下载CUDA

2、 cuDNN下载

根据下载的CUDA版本选择cuDNN下载,CUDA版本一定要和cuDNN版本想对应 

进入到NVIDIA官网,下载对应的cuDNN

https://developer.nvidia.com/rdp/cudnn-archive

选择cuDNN v8.1.0版本然后选择cuDNN Library for Windows 10

首次下载cuDNN要先使用邮箱注册,注册完在重新选择下载即可 

3、CUDA安装

使用默认安装路径就可以,一定要安装在c盘

同意下一步

选择自定义安装

 

后面安装选项默认即可

4、cuDNN

将下载的压缩包解压,并将文件夹中的所有内容复制到cuda的文件夹,替换同名文件夹

 

CUDA和cuDNN安装完成

5、安装pytorch

 (1)打开anaconda prompt

 (2)创建一个使用pytorch的新环境

conda create -n pytorch-gpu python=3.7

(3)进入环境

根据提示进入环境

conda activate pytorch-gpu

 

 

(4)进入pytorch官网查看安装命令 

https://pytorch.org/get-started/previous-versions/

一定要和CUDA版本对应,否则会出现问题 

pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

(5)安装完成以后验证是否安装成功

import torch
torch.__version__
torch.cuda.is_available()

结果返回true表示安装成功

6、torch-geometric安装

在pytorch官网找到下载地址

Installation — pytorch_geometric documentation

官网只有最新的两个版本的pytorch下载,因此找到历史版本,下载

点击torch-1.8.1+cu102,然后,选择参数与python版本相同的四个文件进行下载,其中cp37表示python版本3.7

然后将下载好的文件放入到anaconda的Scripts文件夹中

在环境下安装torch-geometric

pip install torch-geometric

 使用list命令查看

conda list | findstr torch

 

 

 

 

 

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/qq_45137760/article/details/129058365

智能推荐

centos7编译grpc源码_centos grpc 怎么编译-程序员宅基地

文章浏览阅读92次。centos升级gcc v7:_centos grpc 怎么编译

TIM输出比较 P2 PWM控制呼吸灯-程序员宅基地

文章浏览阅读679次,点赞29次,收藏5次。我的理解是xO1 = xIN1 & PWMx;引入PWMx可以更方便的控制特定的电路。四、函数学习/*****单独设置输出比较极性*****///带N的就是高级定时器里互补通道的配置//OC4没有互补通道//结构体初始化也可以设置极性,这里就是单独更改输出极性的,两者效果一样/*****单独修改输出使能参数*****//*****单独更改输出比较模式*****/

少儿编程 2023年12月电子学会图形化编程等级考试Scratch二级真题解析(选择题)_推箱子地图第一行111111,第二行-程序员宅基地

文章浏览阅读151次。​2023年12月scratch编程等级考试二级真题选择题(共25题,每题2分,共50分)1、在制作推箱子游戏时,地图是用数字形式储存在电脑里的,下图是一个推箱子地图,地图表示如下:第一行( 111111)第二行( 132231) 第三行( 126621) 第四行( ) 第五行( 152321) 第六行( 111111 ) 根据规律,第四行应该用哪些数字表示A、( 152121 )B、( 162121 )C、( 122121 )D、( 132121 )答案:B考点分析:考查_推箱子地图第一行111111,第二行

Halcon的Hobject和Opencv的Mat数据类型之间的超快速转换方法_halcon和cvmat数据转换-程序员宅基地

文章浏览阅读3.6k次,点赞12次,收藏41次。由于项目需求,需要把部分Opencv代码和Halcon代码联接起来使用,而两种库所使用的基本数据类型是完全不一样的。因此需要一个互转函数来对他们进行转换。我在这篇博客中看到了一种互转方法Halcon 与 OpenCV 图像数据类型转换。这里面所介绍的方法转换速度非常的慢,Halcon转Opencv需要600ms,而opencv转halcon要10ms,这对于工业应用的项目来说实在是太慢了。仔细研读后,发现这份代码使用Halcon的 GetGrayval() 函数来对HImage进行像素值遍历读取,这_halcon和cvmat数据转换

Android手机验证码倒计时的实现_andriod验证码倒计时-程序员宅基地

文章浏览阅读1.1k次。/**获取验证码倒计时 * Created on 2019/7/4. */public class CountDownTimerUtils extends CountDownTimer { private TextView mTextView; public CountDownTimerUtils(TextView textView, long millisInFut..._andriod验证码倒计时

java中的类的继承_Java中类的继承-程序员宅基地

文章浏览阅读175次。一、子类继承父类的写法class 子类名 extends 父类名{}复制代码二、子类与父类之间的关系①父类可以有个子类②子类只能有一个父类③子类会继承父类的一些属性和方法④子类在创建对象时,会先调用父类的构造方法6三、四大权限修饰符继承非继承同一个包不同包同一个包不同包private缺省protectedpublic上面的对错仅代表能否调用。其实子类继承父类之后,创..._7-2 java类的继承-成员与方法 分数 10 作者 李建华 单位 石家庄铁道大学

随便推点

封神台-杂项11:这是我最喜欢的女明星-程序员宅基地

文章浏览阅读322次,点赞7次,收藏11次。打开发现就是一张图片,根据提示这是一个女明星,直接搜狗识图。找到外文名,这就是flag。发现是邱淑贞,直接百科,

IEEE--DSConv: Efficient Convolution Operator 论文翻译_dsconv: effcient convolution operator-程序员宅基地

DSConv是一种高效的卷积操作符,可以替代传统的卷积操作符,具有高准确性、低内存使用和灵活性。通过对CIFAR10图像的分类实验,证明了DSConv的有效性。

Vue进阶(幺陆伍)PhantomJS 实战讲解_vue phantomjs-程序员宅基地

文章浏览阅读440次。在前期博文《Vue进阶(五十六):vue-cli 脚手架 karma.conf.js 配置文件详解》中讲解了文件用于配置前端项目自动化测试信息。// 这里使用的是PhantomJS作为浏览器测试环境,这个插件支持 DOM, CSS, JSON, Canvas, and SVG 的解析其中,使用了PhantomJS作为无头浏览器测试环境。PhantomJS具体作用是什么呢?本篇博文带你一探究竟。_vue phantomjs

CSS3做出条纹大背景-程序员宅基地

文章浏览阅读168次。㈠实现不等宽背景条纹实现如上图所示的效果,代码如下: 1 <!DOCTYPE html> 2 <html lang="en"> 3 <head> 4 <meta charset="UTF-8"> 5 <style type="text/css"> 6 .con..._css3 条纹背景

IT项目实施流程及每个阶段输出的文档_项目管理各个阶段输出文档-程序员宅基地

文章浏览阅读3.3k次,点赞2次,收藏12次。IT项目实施流程及每个阶段输出的文档_项目管理各个阶段输出文档

【瑞萨RA_FSP】SCL UART 串口通信_sci和uart-程序员宅基地

文章浏览阅读1.6k次,点赞3次,收藏11次。串口通讯(Serial Communication)是一种设备间非常常用的串行通讯方式,因为它简单便捷,因此大部分电子设备都支持该通讯方式,电子工程师在调试设备时也经常使用该通讯方式输出调试信息。在计算机科学里,大部分复杂的问题都可以通过分层来简化。如芯片被分为内核层和片上外设。对于通讯协议, 我们也以分层的方式来理解,最基本的是把它分为物理层和协议层。物理层规定通讯系统中具有机械、电子功能部分的特性,确保原始数据在物理媒体的传输。协议层主要规定通讯逻辑,统一收发双方的数据打包、解包标准。_sci和uart

推荐文章

热门文章

相关标签