- 博客(3)
- 收藏
- 关注
转载 PyTorch中使用指定的GPU
PyTorch中使用指定的GPU PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。 有如下两种方法来指定需要使用的GPU。 1.使用CUDA_VISIBLE_DEVICES 1.1 直接终端中设定: CUDA_VISIBLE_DEVICES=1 python main.py 1.2 python代码中设定: import os os.environ["CUDA_VISIBLE_DEVICES"] = "2" 2.使用函数 set_device import torc
2021-01-12 15:08:05
257
原创 Visual Stdio fopen错误
Visual Stdio fopen错误出现问题解决方案 出现问题 将C语言代码从codeblocks复制到Visual Stdio 2013中,生成解决方案时出现 ‘fopen’: This function or variable may be unsafe. Consider using fopen_s instead. 如图所示: 解决方案 项目->train_network属性 配置属性->c/c+±>预处理器 预处理器定义->编辑 输入_CRT_SECURE_NO
2020-12-27 21:29:37
1913
原创 Visual Studio 2013 “***“:重定义;不同的基类型
Visual Studio 2013 "***":重定义;不同的基类型出现问题解决方案原理解释pragma once宏的作用两者有何区别?两者各有何特点?(1)#ifndef(2)#pragma once两者之间有什么联系? 出现问题 使用Visual Studio 2013 生成解决方案时,出现 “***”:重定义;不同的基类型 的问题,如图所示: 解决方案 在定义所在在头文件最上方加上一句: #pragma once 即可编译通过: 原理解释 pragma once宏的作用 为了避免同一个头文件
2020-12-27 21:05:42
3775
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人