首页 > 其他 > 详细

pytorch

时间:2019-06-21 18:54:09      阅读:119      评论:0      收藏:0      [点我收藏+]

安装

#pytorch为环境名,这里创建python3.5版。
conda create -n pytorch python=3.5

# 切换到pytorch环境 activate pytorch # win下查看cuda版本命令nvcc -V # cpu版本 conda install pytorch-cpu torchvision-cpu -c pytorch # GPU版 conda install pytorch torchvision cudatoolkit=10.0 -c pytorch



配置 Jupyter Notebook

#安装ipykernel
conda install ipykernel
#写入环境
python -m ipykernel install  --name pytorch --display-name "Pytorch for Deeplearning"

#切换回基础环境
activate base
#创建jupyter notebook配置文件
jupyter notebook --generate-config
## 这里会显示创建jupyter_notebook_config.py的具体位置

打开文件,修改

c.NotebookApp.notebook_dir = ‘‘ 默认目录位置
c.NotebookApp.iopub_data_rate_limit = 100000000 这个改大一些否则有可能报错
 

 

pytorch

原文:https://www.cnblogs.com/crazybird123/p/11066237.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!