pytorch
fourye007
Work for fun, Live for love
展开
-
torch: pin-memory, page-lock
对CUDA架构而言,主机端的内存被分为两种,一种是可分页内存(pageable memroy)和页锁定内存(page-lock或 pinned)。可分页内存是由操作系统API malloc()在主机上分配的,页锁定内存是由CUDA函数cudaHostAlloc()在主机内存上分配的,页锁定内存的重要属性是主机的操作系统将不会对这块内存进行分页和交换操作,确保该内存始终驻留在物理内存中。 GPU知道...转载 2019-03-21 11:50:27 · 770 阅读 · 0 评论 -
RuntimeError: Expected object of scalar type Long but got scalar type Double
问题说明 最近在使用一个新的数据集的时候,只改了原先代码的数据路径,和一些简单数据处理代码,在跑的时候就报了如下的错误: Traceback (most recent call last): File "train.py", line 172, in <module> main(parser.parse_args()) File "train.py", line 75,...原创 2019-06-22 13:57:31 · 10300 阅读 · 4 评论