【pytorch】用cuda(GPU)处理数据

目录

  • 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 = 

你可能感兴趣的:(pytorch入门到精通,pytorch,深度学习,python)