[转][PyTorch指定GPU
来源:互联网 发布:nba2k17 mac版本 编辑:程序博客网 时间:2024/05/23 02:28
原文出处:PyTorch指定GPU
PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。
有如下两种方法来指定需要使用的GPU。
1 . 类似tensorflow指定GPU的方式,使用CUDA_VISIBLE_DEVICES。
1.1 直接终端中设定:
CUDA_VISIBLE_DEVICES=1 python my_script.py
1.2 python代码中设定:
import osos.environ["CUDA_VISIBLE_DEVICES"] = "2"
见网址:http://www.cnblogs.com/darkknightzh/p/6591923.html
2 . 使用函数 set_device
import torchtorch.cuda.set_device(id)
该函数见 pytorch-master\torch\cuda__init__.py。
不过官方建议使用CUDA_VISIBLE_DEVICES,不建议使用 set_device 函数。
阅读全文
0 0
- [转][PyTorch指定GPU
- PyTorch使用指定的GPU
- PyTorch 设置使用GPU
- pytorch-multi-gpu
- pytorch-multi-gpu
- ubuntu16.04详细安装pytorch(GPU)
- Ubuntu16.04 anaconda pytorch-gpu环境
- anaconda+pytorch安装(无GPU版本)
- 没有GPU的pytorch安装命令
- caffe 指定GPU
- Keras指定使用GPU
- tensorflow 指定GPU
- Keras指定使用GPU
- Tensorflow 指定GPU运行
- (转)tensorflow中使用指定的GPU及GPU显存
- Pytorch入门学习(四)---- 多GPU的使用
- pytorch multi-process 在 multi-gpu 上的 deadlock
- 浅谈将Pytorch模型从CPU转换成GPU
- thinking in java 14章 类型信息
- 2017.12.4 刘明春第一天总结
- Java反射之获取私有公共的属性和方法
- JavaSE基础知识学习—集合
- 关于keyfind的优化
- [转][PyTorch指定GPU
- jsp、http请求
- Xenomai在Ubuntu中的安装-实时内核更换
- rn中发送跨域的请求
- Jzoj3756 【NOI2014】动物园
- 数据结构实验之图论二:图的深度遍历
- struts2_day03_01_ognl概述_02_ognl入门案例
- 如何破坏单例模式?如何防止?
- 浅谈ECharts3 柱形图使用心得