pytorch环境配置

整理pytorch环境搭建的相关的内容


安装cuda和cudnn

参考链接:cuda安装
如果没有GPU,忽略此步骤

安装anaconda

在anaconda官网下载individual edition

安装pytorch

  1. 添加源:在anaconda prompt中输入如下指令
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  1. 创建pytorch环境:相当于为pytorch新建一个房间。如果不进行这一步的话,就很有可能把pytorch安装在base环境下,虽然也不会报错,但很容易把环境弄乱
conda create -n env_name python=x.x(3.7)

使用 conda env listconda info --envs 查看pytorch环境是否创建成功,创建成功可以看到名为base环境下会列出刚创建的pytorch环境

  1. 激活pytorch环境:完成此步之后才可继续进行下一步安装pytorch,否则pytorch将被装在base环境下
conda activate pytorch
  1. 安装pytorch:在pytorch官网选择电脑的对应配置,复制对应的安装命令
  2. 测试pytorch环境:在命令行中 import pytorch 测试是否安装成功

pycharm中配置pytorch

  1. 添加解释器:settingsproject interpreterAdd

  2. conda environment下找到pytorch虚拟环境
    添加解释器

  3. anaconda下添加的pytorch虚拟环境位置查看 :(base) C:\Users\km>conda env list

  4. 如果报错的话,可以点图中的 Show All,把红的删除即可

jupyter notebook 内核启动时报错

juypter 启动时提示 KernelRestarter: restart failed,猜测应该是版本的问题,折腾一顿之后还是重装了ipykernel可以了

参考链接

  1. Anaconda下的pytorch环境配置
  2. anaconda的下载和安装步骤
  3. Jupyter-notebook报错KernelRestarter: restart failed
  4. jupyter notebook找不到内核解决方法
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值