1. 首页
  2. 编程语言
  3. Python
  4. Pytorch如何切换 cpu和gpu的使用详解

Pytorch如何切换 cpu和gpu的使用详解

上传者: 2021-06-13 00:28:56上传 PDF文件 59.56 KB 热度 33次

前言,在pytorch中,当服务器上的gpu被占用时,很多时候我们想先用cpu调试下代码,那么就需要进行gpu和cpu的切换。在服务器上创建一个python脚本 t.py:首先先看一下,正常运行的情况:

下载地址
用户评论