安装pytorch

下载anaconda

官网下载

生成虚拟环境

conda create --name pytorch python=3.6

pytorch为虚拟环境的名字
虚拟环境的其他操作Anaconda创建虚拟环境及更换pip安装源

安装成功后,输入以下指令:

conda info --envs

即可看到 conda 环境中,有新建的 pytorch 环境,右边的 * 号表示,当前你处于哪个环境。

(base) D:\>conda info --envs
# conda environments:
#
base                  *  D:\Anaconda3
python3                  D:\Anaconda3\envs\python3

切换到虚拟环境

接下来,我们要在 pytorch 环境中,安装 PyTorch,(有点绕),使用如下指令,进入 pytorch 环境。

conda activate pytorch

你可以看到左边的 base 变成了 pytorch,代表成功进入 pytorch 环境。

安装cpu的pytorch

如果你没有gpu环境,想使用Pytorch,打开Pytorch的官网:https://pytorch.org/

在首页我们可以看到有各种配置可选,我们这里选择CPU的版本(CUDA部分选择None):
在这里插入图片描述

安装demo为:

conda install pytorch torchvision cpuonly -c pytorch

或者

pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

import torch报错

如果错误是
No module named ‘torch’

遇到问题,就要解决问题,方法如下:
打开 Anaconda Navigator;
切换到安装 PyTorch 虚拟环境 (如下图):

在这里插入图片描述

3. 安装 Jupyter Notebook(如下图);

在这里插入图片描述

  1. 重启 Joupyter Notebook,即可

import torchvision 报错

如果安装 在 Jupyter 里 import torch 没有问题,可 import torchvision 确报错了:

ImportError: DLL load failed: 找不到指定的模块。

那就在 Anaconda Prompt 里输入以下命令:

conda uninstall pillow
pip install pillow

使用 pip 重新安装 pillow 模块即可。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值