Jupyter Notebook 中运行Pytorch遇到的问题

找不到molde

解决:

打开Anaconda Prompt,激活安装pytorch的那个环境变量,在此环境变量中安装jupyter notebook,命令如下:

conda install ipython
conda install jupyter

在Anaconda Program中运行Pytorch所属的环境,然后

pip uninstall ipykernel

pip install ipykernel

python -m ipykernel install --name pytorch

再打开Jupyter Notebook中new时会出现一个pytorch

同理TensorFlow也一样可以

简单运行实例

import torch
x_data = torch.Tensor([[1.0], [2.0], [3.0]])
y_data = torch.Tensor([[2.0], [4.0], [6.0]])

class LinearModel(torch.nn.Module): 
    def __init__(self):
        super(LinearModel, self).__init__() 
        self.linear = torch.nn.Linear(1, 1)
    def forward(self, x): 
        y_pred = self.linear(x) 
        return y_pred
    
model = LinearModel()
criterion = torch.nn.MSELoss(size_average=False)
optimizer = torch.optim.SGD(model.parameters(), lr=0.01)

for epoch in range(1000):
    y_pred = model(x_data)
    loss = criterion(y_pred, y_data) 
    print(epoch, loss.item())
    
    optimizer.zero_grad()
    loss.backward()
    optimizer.step()
    
print('w = ', model.linear.weight.item())
print('b = ', model.linear.bias.item())
x_test = torch.Tensor([[4.0]]) 
y_test = model(x_test) 
print('y_pred = ', y_test.data)

Anaconda查看、创建、切换虚拟环境

1、查看已有虚拟环境

# 在命令行输入以下命令
conda info --envs

2、创建新的虚拟环境

# 在命令行输入如下命令
conda create --name newName python=3.7

3、切换虚拟环境(非常重要)
当你的本地创建了多个虚拟环境之后,加之又在不同的虚拟环境中安装了不同的第三方package,这样一来,虚拟环境的切换就很重要了。否则,当你再次打开jupyter notebook,通过import命令试图导入某个package的时候,总是会得到如期的错误**Cannot import module. “No module named …”**解决步骤如下:
一、
# 查看已有的虚拟环境,选择你要切换到的虚拟环境
conda info --envs
# 或者
conda env list
二、
# 在命令行中切换到想要的虚拟环境,我这里切换到paddle
conda activate paddle
三、
# 在当前的paddle环境中安装好ipykernel
conda install ipykernel
四、
python -m ipykernel install --name paddle
五、
再次打开jupyter,新建notebook,如图:可以看到,paddle环境出现在了jupyter里面
这样一来,新建的python项目使用的虚拟环境就是paddle。

 

  • 2
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值