PyTorch指定GPU
程序员文章站
2022-06-17 07:59:00
...
PyTorch默认使用从0开始的GPU,如果想用其他GPU,需要指定。
有两种方法可以指定需要用的GPU:
1.类似tensorflow指定GPU的方式,使用CUDA_VISIBLE_DEVICES
1.1 直接终端中设定:
CUDA_VISIBLE_DEVICES=1 python main.py
1.2 python代码中设定:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
2.使用函数 set_device
import torch
torch.cuda.set_device(id)
该函数见 pytorch-master\torch\cuda__init__.py。
不过官方建议使用CUDA_VISIBLE_DEVICES,不建议使用 set_device 函数。
其他详情可参见官方文档:https://pytorch.org/docs/master/cuda.html
https://pytorch-cn.readthedocs.io/zh/stable/package_references/torch-cuda/(中文版)
推荐阅读
-
使用IE9时出现蓝屏或死机通过禁用GPU硬件加速来解决
-
PHP获取指定函数定义在哪个文件中以及其所在的行号实例
-
nginx访问日志并删除指定天数前的日志记录配置方法
-
python输出指定月份日历的方法
-
linux使用tcpdump命令监视指定网络数据包的方法
-
IE9浏览器无法开启GPU硬件加速问题解答
-
记录:Ubuntu 18.04 安装 tensorflow-gpu 版本
-
Ubuntu18.04+cuda10.1+python3.7下配置pytorch0.4.0步骤
-
如何使用Python爬虫获取offcn上的公考信息及写入Excel表格并发送至指定邮箱
-
Nginx屏蔽F5心跳日志、指定IP访问日志