Pytorch
pengjh24
练就有效
展开
-
PyTorch1.6版本之后多卡训练和测试会报错问题解决方案
1.在多进程代码前填写以下这行代码:multiprocessing.set_start_method('spawn')2.若设置后nvidia-smi显示只有一张卡被调用,可考虑将os.environ["CUDA_VISIBLE_DEVICES"]=current_gpu换成torch.cuda.set_device(int(current_gpu))...原创 2021-08-09 17:21:27 · 620 阅读 · 0 评论 -
Pytorch: torch.distributions库
pytorch 的 torch.distributions 中可以定义正态分布:import torchfrom torch.distributions import Normalmean=torch.Tensor([0,2])normal=Normal(mean,1)sample()就是直接在定义的正太分布(均值为mean,标准差std是1)上采样:result = normal...原创 2020-03-18 18:45:20 · 5193 阅读 · 0 评论