首页 > 其他分享 >重新安装CUDA+CUDNN+PyTorch路上遇到的问题

重新安装CUDA+CUDNN+PyTorch路上遇到的问题

时间:2024-07-22 15:53:21浏览次数:15  
标签:重新安装 torch CUDNN PyTorch CUDA https NVIDIA 安装 下载

书接上回,因为CUDA和CUDNN突然用不了了,被迫卸载重新下载。

删除的教程参考的是https://blog.csdn.net/m0_53883779/article/details/135701971。在控制面版卸载中,其实可以留意一下NVIDIA相关软件的安装日期,很快也就分辨出来了,因为我并不是一天把所有的软件都装好了。我这里还有一个NVIDIA USBC Driver,看着安装日期和基础的软件安装日期相同,也没有卸载。来到安装目录,把NVIDIA GPU Computing Toolkit文件夹删除,再用火绒清理一下垃圾,这里卸载就算完成了。

安装的话,按照自己的需求,选择CUDA11.8版本,下载地址:https://developer.nvidia.com/cuda-toolkit-archive。下载完成后,点安装即可,在安装的过程中选择自定义安装,在自定义安装项中,点开CUDA一栏,将Visual Studio Intergration取消,因为我没安装Visual Studio。我按照这个做法很顺利的安装完成了。

安装完后对安装的效果进行检查与验证。查看系统的环境变量,是否生成两个新的系统变量CUDA_PATH和CUDA_PATH_V11_8。打开cmd,输入指令:nvcc --version,查看是否出现CUDA的版本信息。

现在安装CUDNN。首先得注册一个账号,注册的地址:https://developer.nvidia.com/rdp/cudnn-download,页面右上角。找需要的版本地址:https://developer.nvidia.com/rdp/cudnn-archive。下载完成后,解压出来会有bin,include,lib三个文件夹,把他们以管理员身份复制到CUDA的安装目录,我这里是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

安装好以后,打开系统的环境变量,在PATH中添加以下的路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\binC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\includeC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp

添加完成后打开cmd,输入指令:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\extras\demo_suite,分别输入 bandwidthTest.exedeviceQuery.exe ,如果能看到两个PASS,那就算是装好了。

接下来就是最痛苦的一集,装PyTorch。官方地址:https://pytorch.org/get-started/locally/。在上面可以选择符合你设备当前环境的选项,最后会给你显示一个下载命令。一开始我真的以为是这么简单,直接就切换到项目的虚拟环境里面开始运行了,然后这个下载速度令我眉头紧皱,2.5GB的文件正在以100+Kb的速度下载,我真的要裂开了。果断停下来,开一个代理,然后发现用代理下载会被SSL拒绝。没办法,我寻思就这么下吧,好像也不是那么急,过了二十分钟回来一看,连接中断了,我是真的挺无语的,看来又是一番鏖战。

期间去查了很多的方法,发现有很多的方法是在conda的环境下可以运行的,比如说在conda的环境下可以在https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/找到自己所需要的文件,下载完成后直接conda install就完成了。对于上述的方法我试过pip命令运行不了,突然有点羡慕conda。

现在说一说成功的办法,到官方存放资源的网站中去https://download.pytorch.org/whl/torch_stable.html,找到自己所需要的版本。文件名需要看清楚,cu118是CUDA的版本,torch的数字为torch的版本,cp39代表Pyhton3.9,一定要下载对应的torch版本。这里直接下载还是很慢,但是可以用代理,很快就下载好了。用同样的方法也下载好了torchvision,torch和torchvision的对应版本参考以下链接:https://github.com/pytorch/vision

下载完成后安装,转移到虚拟环境中,输入指令:pip install "文件所在的地址\torch-2.3.1+cu118-cp39-cp39-win_amd64.whl",注意这个双引号是一定要的。安装完成后进行验证,在虚拟环境中进入Python,输入指令:

点击查看代码
import torch
torch.cuda.is_available()
若返回的结果是True,则证明安装成功了。

经历了好一阵子的折腾,终于装好了,当我很高兴的准备模型训练时,不出意外就要出意外了,得闲再说。

标签:重新安装,torch,CUDNN,PyTorch,CUDA,https,NVIDIA,安装,下载
From: https://www.cnblogs.com/fuzera/p/18316169

相关文章

  • 基于pytorch演练线性回归模型
    引言本文的目的是在前文基于numpy演练可视化梯度下降的代码基础上,使用pytorch来实现一个功能齐全的线性回归训练模型。为什么仍然使用线性回归模型?线性回归模型简单,它能让我们聚集在pytorch是如何工作的,而不是模型内部的某个复杂结构或算法。与前面的[基于numpy的线性回......
  • pytorch中MultiScaleRoIAlign及MultiScaleRoIPooling实现
    文章目录ROIpooling及ROIAlign原理介绍ROIPoolingROIAlign代码解析使用方式MultiScaleRoIOperation代码解析MultiScaleRoIPooling代码解析MultiScaleRoIAlign代码解析结果引用ROIpooling及ROIAlign原理介绍ROIPoolingRoIPooling用于将任意尺寸感兴趣区域......
  • pytorch学习(八)Dataset加载分类数据集
    我们之前用torchvision加载了pytorch的网络数据集,现在我们用Dataset加载自己的数据集,并且使用DataLoader做成训练数据集。图像是从网上下载的,网址是点这里,标签是图像文件夹名字。下载完成后作为自己的数据集。1.加载自己的数据集的思路  1)要完成继承自Dataset的类的构......
  • 【PyTorch】图像多分类项目
    【PyTorch】图像二分类项目【PyTorch】图像二分类项目-部署【PyTorch】图像多分类项目【PyTorch】图像多分类项目部署多类图像分类的目标是为一组固定类别中的图像分配标签。目录加载和处理数据搭建模型定义损失函数定义优化器训练和迁移学习用随机权重进行训......
  • 安装pytorch(torch)
    建议换源,可以下载速度很快,这里使用的是清华源需要GPU训练,如果没有,自行选择英伟达版本下载修改源,C盘——>user(用户)——>个人用户(自己命名的用户名),找到.condarc,用记事本打开,修改为:channels:-defaultsshow_channel_urls:truedefault_channels:-https://mirro......
  • PyTorch 中 nn.Conv2d()
    nn.Conv2d() 是PyTorch中用于定义二维卷积层(Convolutionallayer)的函数,它属于 torch.nn 模块,该模块包含了构建神经网络所需的所有构建块。二维卷积层是卷积神经网络(CNN)中最基本也是最重要的组件之一,广泛用于图像和视频处理、自然语言处理等领域。nn.Conv2d() 函数的基本语......
  • PyTorch 中 loss.grad_fn 解释
    在PyTorch中,loss.grad_fn属性是用来访问与loss张量相关联的梯度函数的。这个属性主要出现在使用自动微分(automaticdifferentiation)时,特别是在构建和训练神经网络的过程中。当你构建一个计算图(computationalgraph)时,PyTorch会跟踪所有参与计算的操作(比如加法、乘法、激活函数等),......
  • Pytorch的GPU版本安装
    本文的显卡是NVIDIAGeForceRTX3060LaptopGPU,安装环境是CUDA11.1+CUDNN11.1torch1.9.0+cu111torchvision0.10.0+cu111torchaudio==0.9.0 在安装之前先要知道自己的显卡支持什么CDUA版本右键点击进入NVIDIA控制面板选择帮助——系统信息——组件这里可以看出......
  • PyTorch和CUDA版本对应关系
    转自:截至2022.8.19结论:10.2和11.3能兼容大部分版本的pytorch官网链接:https://pytorch.org/get-started/previous-versions/注意:注意低版本的pytorch是否支持更高版本的cuda。(高版本的pytorch一般能兼容低版本cuda)例如:你需要1.7.0的pytorch,那么cuda只能11.0及以下。官方......
  • 基于 CNN(二维卷积Conv2D)+LSTM 实现股票多变量时间序列预测(PyTorch版)
    前言系列专栏:【深度学习:算法项目实战】✨︎涉及医疗健康、财经金融、商业零售、食品饮料、运动健身、交通运输、环境科学、社交媒体以及文本和图像处理等诸多领域,讨论了各种复杂的深度神经网络思想,如卷积神经网络、循环神经网络、生成对抗网络、门控循环单元、长短期记忆......