第一步：下载cuda和cudnn

1.首先要查看自己电脑英伟达gpu的版本从而下载相应的cuda版本，有两种方法查看：

①打开cmd命令行：输入nvcc --version

例如我的版本就是11.6的

②进入控制版面-->硬件和声音-->NVIDIA控制面板-->帮助点系统信息-->点击组件即可看到

2.cuda下载地址：CUDA Toolkit Archive | NVIDIA Developer

下载对应版本的cuda即可

选择相应的版本下载即可：

3.下一步下载cudnn，链接：

cuDNN Archive | NVIDIA Developer

选择对应版本：

点击后要注册或登录英伟达账户才能下载，很麻烦，这里可以右键复制下载链接到迅雷直接下载更方便

第二部：安装CUDA、cudnn和配置环境变量

1.安装cuda:

双击运行下载好的cuda的exe文件，安装时不要乱改路径，一路OK，同意等待就好了

选择自定义安装

后面的路径也不要改，直接安装即可。安装完成后cmd命令行输入“nvcc -V”，查看是否成功

例如这样就可以了：

2.轮到cudnn了

将下载好的压缩包解压后重命名里面的文件名为“cudnn”，然后再复制到该路径下

接着不要忘记配置环境变量（我曾两度忘记）

找到控制面板-->系统和安全-->系统（或进入我的电脑，右键选择属性）选高级系统设置找到环境变量

进入path这个变量

新建，添加这两个路径即可

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\cudnn\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\extras\CUPTI\lib64

第三步：下载cudatoolkit和GPU版本的pytorch、pytorchvision

1.cudatoolkit下载链接：Start Locally | PyTorch

选择对应版本的，复制命令取cmd里执行即可（第一次有部分包安装失败可以多尝试几次）

下载完在cmd里输入“conda list”查看，有cudatoolkit即可

2.GPU版本的pytorch、pytorchvision的下载链接：https://download.pytorch.org/whl/torch_stable.html

里面的文件是cpu开头的是CPU版本，cu才是我们要下的gpu版本

我的cuda版本比较高，下的是1.9.1的torch版本（cuda版本对应的具体torch版本可以百度）；我的python是3.9的，对应的是后面的39；区别win和linux

torchvision也一样

下载好后放在同一个目录下，在cmd下进入这个路径，pip install +“文件名“（注意有引号）

(cmd进入指定目录的方法：①输入”盘符+ : “②输入”cd+空格+指定路径+回车“)

查看检验方法：输入命令：pip list,看到一下就好了

Ture说明成功了。

GPU版本pytorch安装教程相关推荐

TensorFlow2.0 系列开篇： Windows下GPU版本详细安装教程
点击上方"Datawhale",选择"星标"公众号第一时间获取价值内容 [导读]今年三月谷歌在TensorFlow开发者峰会上宣布TensorFlow 2.0 ...
旧版本pytorch安装教程
目录 Anaconda 创建虚拟环境 pytorch官网下载 Anaconda 首先要有这个,然后建立虚拟环境----->进入虚拟环境创建虚拟环境 1.利用Anaconda Navigato ...
win11 系统 GPU版本pytorch、CUDA、anaconda 、pycharm详细安装教程
win11 系统深度学习环境搭建----- GPU版本pytorch.CUDA.cuDnn.anaconda .tensorflow_gpu.pycharm详细安装教程参考比站视频: PyTorch ...
Ubuntu无root权限安装miniconda到配置jupyter、GPU版本pytorch一条龙服务
前言倒腾学校服务器来回几次了,对学校给学生无root普通用户权限,老师要求的只能安装miniconda然后手动配jupyter.GPU版本pytorch等环境已经轻车熟路了. 如果你是老手,完全可以 ...
Pytorch安装教程及解决 torch.cuda.is_available() 返回 False 的问题
Pytorch安装教程及解决 torch.cuda.is_available() 返回 False 的问题有NVIDIA显卡的可以先去看CUDA安装教程,pytorch可以搭配CUDA10.0~ ...
vs最好的版本_Win10 环境下，LightGBM GPU 版本的安装
由于知乎的编辑器不能完全支持 MarkDown 语法, 所以部分文字可能无法正常排版, 如果你想追求更好的阅读体验, 请移步至该博客的简书的链接. Win10 平台下, LightGBM GPU 版本 ...
Windows下显卡支持的cuda及配套GPU版本pytorch下载全流程（2022.03.27）
Windows下显卡支持的cuda及配套GPU版本pytorch下载全流程(2022.03.27) 一.更新电脑显卡驱动(以mx150为例) 1.打开英伟达驱动下载官网:https://www.nvi ...
Win10 平台下, LightGBM GPU 版本的安装
1. Light GBM 简介在数据挖掘和传统机器学习领域,提起大名鼎鼎的 XGBoost,相信很多人都听说过,尤其是在 Kaggle 赛场上,XGBoost 更是风光无限,不过今天的主角却不是 X ...
Pycharm和Pytorch安装教程配置环境以及遇到的问题：
Pycharm和Pytorch安装教程配置环境以及遇到的问题: 注意:我们每次新建完项目,都要检查一下python解释器和conda.exe是否选择正确. 一.如何找到Anconda哪个环境中安装了p ...

GPU版本pytorch安装教程

第一步：下载cuda和cudnn

第二部：安装CUDA、cudnn和配置环境变量

第三步：下载cudatoolkit和GPU版本的pytorch、pytorchvision

GPU版本pytorch安装教程相关推荐

最新文章

热门文章