pytorch实战教程
SimonChenHere
PhD Candidate,计算机本,人工智能硕,网络安全博。
展开
-
无法安装 torch-sparse
Collecting torch-sparse Using cached torch_sparse-0.6.0.tar.gz (29 kB)Requirement already satisfied: scipy in c:\users\caleb\anaconda3\envs\graphstar\lib\site-packages (from torch-sparse) (1.4.1)Requirement already satisfied: numpy>=1.13.3 in c:\use原创 2021-07-06 19:08:03 · 1051 阅读 · 0 评论 -
技术迭代快速。PyTorch 真的优于Tensorflow吗?
pytorchpytorch形式上高度面向对象,tf则保留了许多面向过程的特性。Research的目的是快速实现、验证自己的想法,而不太注重兼容、部署等问题运行环境低依赖,cuda is all you need,别人的工作克隆下来+anaconda就能直接跑,高效。tfTensorFlow强在在线部署,多语言支持和较好的线上系统稳定性,我觉得这个是TensorFlow在业界被广泛应用的最主要原因目前有些公司放弃了tf,用pytorch。 - 分布式TF很不好用。 - Ref:ht原创 2021-01-20 20:41:21 · 215 阅读 · 0 评论 -
argument ‘input‘ (position 1) must be Tensor, not XX
这里:numpy 和 tensor数据相互转换的方法:1.numpy->tensordata = torch.tensor(data)2.tensor->numpy非训练数据(训练结束后的tensor)的转换:data1 = np.array(data1)如果是训练过程中需要转换,则:data_ = data.cpu().data.numpy()https://blog.csdn.net/qq_41368074/article/details/105560274...原创 2020-11-06 19:04:58 · 35327 阅读 · 0 评论 -
ImportError: cannot import name ‘Optional‘
查看自己的pytorch版本和torchvision版本是否对照~安装pytorch1.0 要对应安装torchvision0.2.0https://blog.csdn.net/qq_41558173/article/details/106101164原创 2020-10-12 16:32:01 · 654 阅读 · 0 评论 -
pytorch教程:Flatten类实现的模版
大多数可以套用:原创 2020-09-21 08:25:06 · 433 阅读 · 0 评论 -
pytorch教程:save and load
torch有一个“save and load”机制,即训练意外终止了,可以保存训练的中间文件(ckpt.mdl),然后恢复训练的最开始重新加载进来。原创 2020-09-21 08:19:34 · 915 阅读 · 0 评论 -
torch验证是否在使用GPU
import torcha = torch.cuda.is_available()print(a)ngpu= 1# Decide which device we want to run ondevice = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")print(device)print(torch.cuda.get_device_name(0))print(torch.ra原创 2020-09-19 18:34:42 · 5438 阅读 · 0 评论