python windows安装cuda+cudnn+pytorch教程
作者:L'y 发布时间:2023-02-04 04:35:02
安装cuda 我强调下 这个需要注意版本问题的
注意 (个人的想法,安装思路,仅供参考)
pytorch 需要注意这个现在支持的版本.根据这个支持的版本去下载对应的cuda和cudnn
应为你 pytorch 的版本对不上 你cuda装上了也不行的. 到时候检测运行的时候直接就False.
我现在安装的pytorch
支持 cuda
的版本是 11.3. 所以你们安装的时候需要注意. 应该是可以用低版本的安装的. (指的是cuda). 切记,切记,切记,
https://pytorch.org/ 这是 pytorch
的官网.
简单的说
先看 pytorch 官网是支持的cuda的版本.
然后去 cuda下载网站 查看对应的版本,(自己看着来,我下载的时候是对着pytorch上面需要的版本来的) ,注意自己的显卡驱动的版本.不要低于cuda要求的显卡版本.
对应 cuda的版本去下载cudnn
都安装配置完成后,再去pytorch 官网下载 pytorch 就行 直接复制下载即可.
最后运行测试下.
以下出现的所有cuda和cudnn的版本都是作为安装教程使用,具体版本根据情况而定.
以下出现的所有cuda和cudnn的版本都是作为安装教程使用,具体版本根据情况而定.
以下出现的所有cuda和cudnn的版本都是作为安装教程使用,具体版本根据情况而定.
图片只是记录一个安装流程.
先校验英伟达显卡版本.(后续切勿自动更新这个)
对应显卡版本进行下载对应的cuda版本 (也可以降低版本安装)
这是网址 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
找到自己合适的版本即可.
安装的版本 不能超过pytorch限制的版本
简单的说下这里的逻辑:
pytorch版本限制你cuda的版本
cuda限制你的显卡驱动版本
所以说 pytorch要求的cuda的版本时, 需注意自己显卡驱动的版本不得低于cuda要求版本.
举例子 pytorch 安装 11.3版本的 cuda . 11.3版本的cuda 需要显卡驱动为 469.xx的.所以说 我们需要显卡版本是 大于等于 469.xx的.
下载安装 cudaa
下载链接 https://developer.nvidia.com/cuda-toolkit-archive
下载自己合适的版本.
我的是windows10 我选择本地安装.
进行安装.
文件存放的地方
需要安装的模块
模块安装的位置
安装完成之后 查看下自己环境变量
这样基本上就ok了
简单的测试下
打开cmd终端 ,执行命令
nvcc -Vnvcc --version
这样就ok了
下载 cudnn
下载网址 https://developer.nvidia.com/rdp/cudnn-archive
下载合适自己的版本
将cudnn文件进行解压 放到cuda的目录下.
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4
这里我的路径 每个人的路径可能不一样.
测试使用下.
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\extras\demo_suite
找到这个demo_suite
这个文件夹
然后调用这个exe.
效果如下就基本ok了
官网下载安装 pytorch https://pytorch.org/
import torch
print(f'torch -v {torch.__version__}')
print(f'cuda is_available {torch.cuda.is_available()}')
结果:
torch -v 1.10.1+cu113
cuda is_available True
来源:https://blog.csdn.net/Ly_Word/article/details/122166252
猜你喜欢
- 一、 [::-1]import numpy as npimport numpy as npx = np.arange(1, 6)print(
- Python import .pyd文件时会搜索sys.path列表中的路径运行import xxx.pyd1. 'ImportEr
- 简单低级的爬虫速度快,伪装度低,如果没有反爬机制,它们可以很快的抓取大量数据,甚至因为请求过多,造成服务器不能正常工作。而伪装度高的爬虫爬取
- 通过锁机制,可以实现多线程同时对某个表进行操作。如下图所示,在某个时刻,用户甲、用户乙、用户丙可能会同时或者先后(前面一个作业还没有完成)对
- import pdbdef pdb_test(arg): for i in range(arg):&nb
- Python中核心的数据类型有哪些?变量(数字、字符串、元组、列表、字典)什么是数据的不可变性?哪些数据类型具有不可变性数据的不可变是指数据
- 如下所示:import pandas as pdcontent = ['T', 'F'] * 10data
- 源码:#路飞骷髅import turtle as t#黄底帽子t.pu()t.goto(0,200)t.circle(-130,-80)t.
- 本文实例讲述了PHP动态生成指定大小随机图片的方法。分享给大家供大家参考,具体如下:<?php$image_width = 100;$
- flatten()函数用法flatten是numpy.ndarray.flatten的一个函数,即返回一个一维数组。flatten只能适用于
- 近日,有朋友一直打听flash连结服务器相关的知识,搞得我忧心重重,重点是自己也忘记了,大部分Flash的相关开发都是两年前的事,而且fla
- 1 模型定义和TF很像,Pytorch也通过继承父类来搭建模型,同样也是实现两个方法。在TF中是__init__()和
- django模板使用media文件夹,想要在前端通过{{ MEDIA_URL }}无法显示图片,没有取到Media_url的值解决办法:TE
- php的引用(就是在变量或者函数、对象等前面加上&符号),在PHP 中引用的意思是:不同的名字访问同一个变量内容。与C语言中的指针是
- 在ACCESS中更改控件的默认属性 Lisa Friedrichsen, 欧弗兰帕克,堪萨斯州 如果您在设计一个Microsoft ACCE
- 本文实例讲述了php逐行读取txt文件写入数组的方法。分享给大家供大家参考。具体如下:假设有user.txt文件如下:user01user0
- 这篇文章主要介绍了python 采用paramiko 远程执行命令及报错解决,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的
- Jon Wiley, User Experience Designer for Google Apps, outlined some of
- 现在网页的设计都讲究整体统一风格,无论是网页的文字、图像,还是浏览器的滚动条都要求颜色和风
- 在创建SQL Server 2000 故障转移群集之前,必须配置 Microsoft 群集服务 (MSCS) 并使用 Microsoft W