pytorch的cuda环境搭建（GPU版本安装）

第一步：安装cuda

安装cuda，这里还需要安装对应版本的Visual Studio（具体参考本博主的博客 https://blog.csdn.net/qq_36653505/article/details/81368346 ）

第二步：安装cudnn

cudnn是pytorch搭建深度学习模型的依赖，没有它，不能运行卷积等操作。
下载对应版本的cudnn链接：https://developer.nvidia.com/rdp/cudnn-archive

打开压缩包，依次将压缩包中的3个文件复制到相应位置

C:\cuda\bin\cudnn64_7.dll —> C:\Program Files\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\bin
C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\include
C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPUComputing
Toolkit\CUDA\v9.1\lib\x64

注意： 一定要安装cudnn，不然无法进行卷积操作

安装GPU版本的pytorch

到官网下载对应GPU版本的pytorch

然后使用pip install 文件名安装下载好的pytorch压缩包

第四步：检查是否安装成功

参考

https://blog.csdn.net/MarsLee_U/article/details/80489618
https://blog.csdn.net/u011622208/article/details/80275344

pytorch的cuda环境搭建（GPU版本安装）相关推荐

Pytorch：入门指南和 PyTorch 的 GPU版本安装(非常详细)
Pytorch: 入门指南和 PyTorch 的 GPU版本安装(非常详细) Copyright: Jingmin Wei, Pattern Recognition and Intelligent S ...
Windows下显卡支持的cuda及配套GPU版本pytorch下载全流程（2022.03.27）
Windows下显卡支持的cuda及配套GPU版本pytorch下载全流程(2022.03.27) 一.更新电脑显卡驱动(以mx150为例) 1.打开英伟达驱动下载官网:https://www.nvi ...
3050显卡驱动安装+配置pytorch的cuda环境
3050显卡驱动安装+配置pytorch的cuda环境前言一.下载3050驱动二.下载CUDA 二.cuDNN下载三.cuDNN配置四.pytorch环境配置 ①.创建虚拟环境前言因为有 ...
pytorch，torch，torchvision的gpu版本安装避坑
pytorch,torch,torchvision的gpu版本安装避坑在线or离线安装 pytorch torch和torchvision 在线or离线安装最近使用pytorch跑深度学习模型,需 ...
TensorFlow GPU 版本安装个人总结：Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0
TensorFlow GPU 版本安装个人总结:Win10 + Python3.5 + CUDA 9.0.176 + cudnn v7.5.0.56 + TensorFlow 1.12.0 接触机器学 ...
mysql的安装版本有无开发版_moodle 2.0.2 安装图文教程 2——PHP Mysql环境搭建和moodle安装 - 乔木的日志 - ICT教育 - 中小学信息技术教育和教育信息化网站...
moodle 2.0.2 安装图文教程 2--PHP+Mysql环境搭建和moodle安装热度 6已有 211 次阅读 2011-3-29 16:49 ||关键词:moodle 2 图文安装教程 p ...
【教程】Windows下Miniconda+Pytorch+Pycharm开发环境搭建指南
Windows下Miniconda+Pytorch+Pycharm开发环境搭建指南给本科生写的新手教程,遇到很多新手的共性问题,可供大家参考.在开始阅读之前,请注意两个要关键:1.为什么要使用Con ...
Windows10 配置 Nvidia 驱动与 Cuda 环境搭建
title: Windows10 配置 Nvidia 驱动与 Cuda 环境搭建 mathjax: false date: 2021-04-17 12:50:27 tags: [Environment ...
深度学习工作站环境搭建 ubantu16.04 安装显卡驱动 CUDA8.0 cuDNN6 anaconda tensorflow1.4.0 opencv3.4 含泪总结
深度学习工作站环境搭建 ubantu16.04 安装显卡驱动 CUDA8.0 cuDNN6 anaconda tensorflow opencv 1 安装ubantu16.04 安装时不要在线更新,在 ...