摘要: pytorch数据类型 python数据类型 Int float Int array Float array torch数据类型 tensor of size() FloatTensor of size() Int array IntTensor of size [d1, d2 ,…] Float 阅读全文
posted @ 2020-04-11 23:53 wqbin 阅读(913) 评论(0) 推荐(0) 编辑
摘要: 好久没有更新pytorch,这次更新做个笔记 1.安装CUDA https://developer.nvidia.com/cuda-toolkit-archive 设置环境变量后输入nvcc -V C:\Users\WQBin>nvcc -V nvcc: NVIDIA (R) Cuda compil 阅读全文
posted @ 2020-04-11 11:44 wqbin 阅读(467) 评论(0) 推荐(0) 编辑