美文网首页
Ubuntu18.04下Anaconda+pytorch安装

Ubuntu18.04下Anaconda+pytorch安装

作者: 西部的玫瑰 | 来源:发表于2019-10-28 09:51 被阅读0次

    一、搭建pytorch环境

    conda create -n pytorch python=3.7

    二、安装pytorch

    激活新建的pytorch环境

    conda activate pytorch

    安装最新的pytorch

    pip3 install torch===1.3.0 torchvision===0.4.1 -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

    #从清华镜像源下载安装,经检验十分的快,此处默认是使用 cuda10.1

    pip3 install torch==1.3.0+cu92 torchvision==0.4.1+cu92 -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

    #cuda9.2

    pip3 install torch==1.3.0+cpu torchvision==0.4.1+cpu -f https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

    #cpu版本

    验证输入python 进入

    import torch

    torch.__version__# 得到结果'1.3.0'

    三、配置 Jupyter Notebook

    新建的环境是没有安装安装ipykernel的所以无法注册到Jupyter Notebook中,所以先要准备下环境

    conda install ipykernel#安装ipykernel(还是在pytorch 环境下)

    python -m ipykernel install --user --name pytorch --display-name Pytorch-for-Deeplearning#写入环境

    下一步就是定制 Jupyter Notebook

    conda activate base#切换回基础环境

    jupyter notebook --generate-config#创建jupyter notebook配置文件,这里会显示创建jupyter_notebook_config.py的具体位置。

    打开文件,修改

    c.NotebookApp.notebook_dir = '' 默认目录位置

    c.NotebookApp.iopub_data_rate_limit = 100000000 这个改大一些否则有可能报错

    至此 pytorch的开发环境安装完成,可以在开始菜单中打开Jupyter Notebook 在New 菜单中创建文件时选择Pytorch-for-Deeplearning 创建pytorch的相关开发环境了,输入如下语句,验证可得即安装成功。

    配置pycharm环境即将下图的内核换为envs/pytorch环境下的python内核即可。

    如有遇到安装问题即可给我留言。

    相关文章

      网友评论

          本文标题:Ubuntu18.04下Anaconda+pytorch安装

          本文链接:https://www.haomeiwen.com/subject/yzevvctx.html