小编Mel*_*ike的帖子

nn.MaxPool2d 与 nn.functional.max_pool2d 之间的区别?

nn.MaxPool2d(kernel_size, stride)和之间有什么区别nn.functional.max_pool2d(t, kernel_size, stride)

我在模块中定义的第一个和转发函数中的第二个?

谢谢

python module forward pytorch

4
推荐指数
1
解决办法
4872
查看次数

Pytorch 无法在 PyCharm/IntelliJ 中与 CUDA 配合使用

我刚刚通过 Anaconda 下载了带有 CUDA 的 PyTorch,当我在 Anaconda 终端中输入:

import torch
if torch.cuda.is_available():
    print('it works')
Run Code Online (Sandbox Code Playgroud)

然后他输出;这意味着它有效并且可以与 PyTorch 一起使用。

但是当我进入 IDE(PyCharm 和 IntelliJ)并编写相同的代码时,它不会输出任何内容。

有人可以向我解释一下如何让它在 IDE 中运行吗?

python intellij-idea pycharm pytorch

3
推荐指数
1
解决办法
2万
查看次数

标签 统计

python ×2

pytorch ×2

forward ×1

intellij-idea ×1

module ×1

pycharm ×1