pytorch
B_storm
准研究生。补基础。
展开
-
Libtorch问题记录Error opening images file at /yourDIR/data/train-images-idx3-ubyte
报错内容CUDA is available! Training on GPU.terminate called after throwing an instance of 'c10::Error' what(): Error opening images file at ./data/train-images-idx3-ubyte (read_images at /yourDIR/torch/csrc/api/src/data/datasets/mnist.cpp:66)frame #0: c1原创 2022-03-03 10:06:46 · 886 阅读 · 0 评论 -
用anaconda安装gpu版pytorch的include文件所在路径
问题描述使用anaconda安装pytorch后,写c++ extension时使用的"cuda.h"和"cuda_runtime.h"的默认包含路径为"/usr/include",与实际使用的版本不符。解决方案真实路径为"/usr/local/cuda-10.0/include/"。其中的"cuda-10.0"替换为自己的版本。...原创 2021-11-23 14:32:20 · 1518 阅读 · 0 评论