Ubuntu18.04下Anaconda+pytorch安装

阿里云2000元红包!本站用户参与享受九折优惠!

一、搭建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内核即可。

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

https://www.jianshu.com/p/c10f347422d7

Python量化投资网携手4326手游为资深游戏玩家推荐:《元气骑士:花园能否扩建,道具只增9种?深耕牙膏更新的隐藏讯息

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论