首页 › Python › 正文

Python pytorch入门到精通深度学习

【pytorch】用cuda（GPU）处理数据

学渣渣渣渣渣 Python 2023-03-17 22 0 原文

目录

1 设置GPU的一些操作
- 避坑！！！ RuntimeError: No CUDA GPUs are available
2 直接在gpu创建
3 从cpu转移到gpu
4 在cuda中训练模型
5 显存不够的问题
- 5.1 使用with torch.no_grad() & torch.cuda.empty_cache()
6 训练时中遇到的卡住停住的问题

考虑到各种运算只能在cpu或者gpu运算，不能混和运算，本文介绍常用的几种把数据挪到gpu或者直接在gpu创建数据再进行运算的方法。

1 设置GPU的一些操作

设置在os端哪些GPU可见，如果不可见，那肯定是不能够调用的~

import os
GPU =

喜欢 (0)

【pytorch】torchvision.transforms 图像的变换详解；图像的预处理；数据增强

【pytorch】卷积层计算规则

发表评论点击这里取消回复。