Debian9.8 安装 pytorch

65 阅读 0 评论 43 点赞

我是靠谱客的博主含蓄冷风，最近开发中收集的这篇文章主要介绍Debian9.8 安装 pytorch，觉得挺不错的，现在分享给大家，希望可以做个参考。

2019独角兽企业重金招聘Python工程师标准>>>

Debian9.8 安装 pytorch (不安装cuda):
apt install -y git python3-dev
git clone --recursive https://github.com/pytorch/pytorch

export NO_CUDA=1

export USE_NUMPY=1

export USE_FFMPEG=1

export USE_OPENCV=1

export USE_REDIS=1
cd pytorch

pip install opencv-python numpy scipy tensorflow scikit-learn matplotlib redis ffmpeg

pip install pyyaml

pip install -r requirements.txt
python3 setup.py install

pip install torchvision

或者直接按照pytorch官网上https://pytorch.org/选择你要安装的，如：

pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp35-cp35m-win_amd64.whl

用命令查看pytorch适合安装成功：

pip list | grep torch

简单例子：

import torch

x = torch.empty(5, 3)
print(x)

输出：

哈哈，以后就可以用pytorch开发了

转载于:https://my.oschina.net/michaelshu/blog/3021075

以上就是含蓄冷风为你收集整理的Debian9.8 安装 pytorch的全部内容，希望文章能够帮你解决Debian9.8 安装 pytorch所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。