WIN10环境下深度学习中使用torch时无法启用GPU的一些坑

使用DeepCTR-zorch来进行深度学习时，

torch.cuda.is_available()

返回值总是false而导致无法使用GPU来进行计算。这对笔者这拥有号称卡皇1080Ti的人来说，简直太不爽了。于是网上找了找资料，发现都介绍的不全面，所以在看本文的时候，当你尝试了网上所说的驱动版本问题，CUDA版本问题，环境变量设置问题以及各种莫名其妙的版本问题后再来看本文，本文假定您基本已安装了其他所有该依赖的相关文件。

先说本文所使用的的环境

Python 3.9

CUDA V11.3

下载链接：https://developer.nvidia.com/cuda-toolkit-archive

cuDNN

下载链接：https://developer.nvidia.com/rdp/cudnn-download

下载完后注意依据文档，直接将解压后的三个目录，直接拷贝覆盖到CUDA安装路径下的同名目录即可。

===========================分割线===========================

Anaconda3-2021.05-Windows-x86_64

下载地址：https://www.anaconda.com/distribution/

Miniconda3-py39_4.9.2-Windows-x86_64

下载地址：https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/

最后，使用conda安装tensorflow的GPU版本，注意不要使用PIP，可能引起版本冲突。

conda install tensorflow-gpu

OK，现在返回值是True了，再次跑一次程序，终于，GPU工作了

WIN10环境下深度学习中使用torch时无法启用GPU的一些坑相关推荐

深度学习中高斯噪声：为什么以及如何使用
在数学上,高斯噪声是一种通过向输入数据添加均值为零和标准差(σ)的正态分布随机值而产生的噪声. 正态分布,也称为高斯分布,是一种连续概率分布,由其概率密度函数 (PDF) 定义: pdf(x) = ( ...
深度学习中：epoch、batch size和iterations之间的关系
机器学习中:epoch.batchsize和iterations之间的关系在深度学习中,epoch.batchsize和iterations是在训练模型时,一定用到的一些概念.那么分别是什么意思呢? ...
Win10配置pytorch深度学习环境
一.Win10配置pytorch深度学习环境有一台装win10系统的电脑有两张显卡,所以尝试在win10上装环境了,之前在Ubuntu上安装过深度学习环境,相比起来还是Ubuntu的命令行配环境更方 ...
win10系统的深度学习环境搭建以win10+rtx2060+tensorflow为例/K210的win10系统的深度学习环境搭建/有无显卡均可安装
win10系统的深度学习环境搭建以win10+rtx2060+tensorflow为例 K210的win10系统的深度学习环境搭建有无显卡均可安装一软件准备 1.Anaconda3 软件介绍:A ...
深度学习中的知识蒸馏技术(下)
本文概览: 写在前面: 这是一篇介绍知识蒸馏在推荐系统中应用的文章,关于知识蒸馏理论基础的详细介绍,请看我的这篇文章: 深度学习中的知识蒸馏技术(上) 1. 背景介绍 1.1 简述推荐系统架构如果从 ...
CV：Win10下深度学习框架安装之Tensorflow/tensorflow_gpu+Cuda+Cudnn(最清楚/最快捷)之详细攻略(图文教程)
CV:Win10下深度学习框架安装之Tensorflow/tensorflow_gpu+Cuda+Cudnn(最清楚/最快捷)之详细攻略(图文教程) 导读本人在Win10下安装深度学习框架Tenso ...
【技术综述】深度学习中的数据增强（下）
文章首发于微信公众号<有三AI> [技术综述]深度学习中的数据增强(下) 今天带来深度学习中的数据增强方法的下篇.我们将从以下几个方向入手.1,介绍一下什么是无监督的数据增强方法.2,简单 ...
win10环境下下载安装openpose(only cpu)并在pycharm中运行代码（超详细）
win10环境下下载安装openpose(only cpu)并在pycharm中运行代码(超详细) (一)前言 (二)准备工作 (三)友情提醒 (四)详细安装步骤 1.新建文件夹 2.下载OpenPo ...
gan 总结数据增强_深度学习中的数据增强（下）
言有三毕业于中国科学院,计算机视觉方向从业者,有三工作室等创始人编辑 | 言有三今天带来深度学习中的数据增强方法的下篇.我们将从以下几个方向入手.1,介绍一下什么是无监督的数据增强方法.2,简单 ...

WIN10环境下深度学习中使用torch时无法启用GPU的一些坑

WIN10环境下深度学习中使用torch时无法启用GPU的一些坑相关推荐

最新文章

热门文章