Pytorch GPU 训练环境搭建

最新推荐文章于 2024-06-07 15:33:14 发布
最新推荐文章于 2024-06-07 15:33:14 发布
阅读量607 收藏 1
点赞数
文章标签： pytorch 深度学习 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/hogwarts_ziqi/article/details/126121186
版权
首次使用PyTorch进行模型训练时，发现GPU未被充分利用，经检查发现CUDA配置问题。通过nvidia-smi检测CUDA版本，并在PyTorch官网找到适配的torch torchvision版本。遇到版本不匹配错误，尝试降低PyTorch版本至1.2.0以匹配CUDA 10.0，但发现官方已不支持，最终决定升级CUDA至10.1以匹配更高版本的PyTorch。
摘要由CSDN通过智能技术生成

之前⼀直使⽤ Tensorflow 训练模型，第⼀次训练Pytorch模型的时候，发现速度很慢，仔细观察，发现GPU 内存占⽤为0，基本没有使⽤GPU。

AssertionError: CUDA unavailable, invalid device 0 requested
cuda不可⽤报错，现实没有有效的驱动可使⽤

测试cuda是否配置正确

import torch
print(torch.cuda.is_available())

重新安装cuda

检测本地GPU CUDA版本 nvidia-smi

pip3 install torch1.9.0+cu101 torchvision0.10.0+cu101 torchaudio=0.9.0 -f
https://download.pytorch.org/whl/torch_stable.html

当前 torchvision0.10.0+cu101 版本必须是匹配的。如果版本不匹配，如上⾯的命令，则会出现错误

我们打开网站

最低0.47元/天 解锁文章
测试人子期
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
pytorch GPU环境搭建（win10，python3.6，pycharm，cuda9.2，anaconda3.5.2）
hbu_pig的专栏
08-15 1万+
cuda下载地址：https://developer.nvidia.com/cuda-downloads pytorch离线安装包下载地址：https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/ pycharm下载地址：http://www.jetbrains.com/pycharm/ anaconda下载地址：...
1 PyTorch（GPU版本 ）环境搭建
睿智小码农
04-13 5448
PyTorch（GPU版本 ）环境搭建 Anaconda的安装，安装一路Next即可，不再赘述； Anaconda换源，打开Anaconda Prompt，在base环境中输入下列命令回车即可； conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/an
Pytorch无镜像无梯子安装逃课版
最新发布
qq_62304195的博客
06-07 664
在不进行换源和上梯子的情况下安装pytorch
深度学习------pytorch-gpu环境搭建
m0_47405013的博客
08-03 1328
pytorch环境搭建
python pytorch-GPU 环境搭建 (CUDA 11.2)
m0_58709899的博客
09-06 1万+
如下图所示，为 pytorch 或 tensorflow 的环境搭建逻辑。
添加清华镜像，快速安装pytorch
fangyunfeng的专栏
01-20 1567
** 添加清华镜像，快速安装pytorch 在windows 10下安装pytorch，注意我这里已经提前安装好了aconda3 ，cuda 10.2，cudnn 10.1 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels ...
pytorch简介与环境搭建
04-13
在Python环境中，PyTorch的Tensor操作类似于NumPy，但在GPU上运行时能实现高效的计算，这对于处理大规模数据和训练深度学习模型至关重要。 环境搭建是使用PyTorch的第一步。首先，你需要安装Python。推荐使用Python...
pytorch 多 gpu 训练代码.docx
03-24
本文将详细介绍如何使用 PyTorch 实现多 GPU 训练，包括数据集和模型的准备、多 GPU 环境的搭建、损失函数与优化器的定义、多 GPU 训练过程、以及模型的保存与加载等方面。 #### 二、准备数据集和模型 在开始多 ...
Yolov8使用gpu训练环境搭建教程 训练素材
04-24
对于GPU训练，设置"batch"和"ngpu"参数，使模型能够充分利用多GPU资源。 接下来，准备数据集。COCO128的数据集需进行预处理，包括图像的缩放、归一化和标注信息的转换。使用YOLOv8提供的脚本，如`prepare_data.py`...
Pytorch环境搭建与基本语法
09-16
本文将深入探讨PyTorch的环境搭建和基本语法，帮助初学者理解和应用这一工具。 首先，让我们来看PyTorch的环境搭建。在安装PyTorch时，我们需要考虑操作系统、CUDA版本和Python版本的兼容性。例如，对于一个Windows...
《PyTorch中文手册》学习之路day1-pytorch简介和环境搭建
12-22
环境搭建是使用PyTorch的第一步。对于Windows 10用户，推荐使用Anaconda来管理Python环境。首先，你需要安装Anaconda，然后在Anaconda Prompt中创建一个新的环境，例如命名为`pytorch`，并指定Python版本为3.6。创建...
搭建Pytorch环境(GPU版本，含CUDA、cuDNN)，并在Pycharm上使用（零基础小白向）
yilvyangguang520的博客
10-17 3039
搭建Pytorch环境(GPU版本，含CUDA、cuDNN)，并在Pycharm上使用（零基础小白向）
【Windows】搭建Pytorch环境(GPU版本，含CUDA、cuDNN)，并在Pycharm上使用（零基础小白向）
热门推荐
a922922737的博客
09-11 2万+
此文记录详细且正确的GPU版Pytorch环境搭建过程，同时包括在Pycharm上使用Pytorch的教程（Anaconda环境）。希望此文对读者有帮助！
windows下的pytorch-gpu环境搭建方法
wwwsctvcom的博客
11-06 228
一、配置国内镜像源 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ #必不可少，不然会出现找不到包，或者只有cpu版本的pytorch包 conda config --add channels https://mi
Linux系统搭建GPU加速的PyTorch环境
FanJin的博客
03-29 5896
当前训练CNN最受欢迎的两个框架是TensorFlow和 pyTorch，搭建相应的环境其实不难，博主早先在笔记本的Ubuntu上很快就搭建了pyTorch环境及手写了ResNet 的Demo。但是最近台式机配上比较不错的显卡后，再次搭建带CUDA的 pyTorch开发环境时，遇到一些问题，故简略记录于此。 博主一直使用的是Ubuntu+Windows双系统，此次一并将双系统（Win10+Ubu...
pytorch -gpu 环境配置
m0_51658421的博客
09-26 6074
pytorch -gpu环境配置 包含cuda和cudnn下载配置 完整过程
pytorch--GPU环境配置
qq_43719529的博客
01-08 688
pytorch GPU环境配置
Pytorch 配置 GPU 环境
Henry的博客
02-19 3575
深度学习跑代码的时候，因为简单的操作不适合cpu运行，我们更习惯用GPU加速代码。本章将介绍怎么安装pytorch的gpu环境，以及常见的问题这里提供的版本都是较新的，电脑的硬件跟不上的话，可以选择之前版本的。
pytorch 3D模型训练环境搭建 ubuntu2004
07-11
要在 Ubuntu 20.04 上搭建 PyTorch 3D 模型训练环境，你可以按照以下步骤进行操作： 1. 安装 CUDA 和 cuDNN（如果你的 GPU 支持 CUDA）： - 首先，检查你的 GPU 支持的 CUDA 版本，并从 NVIDIA 官方网站下载对应...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值