ubuntu 安装pytorch

前端之家收集整理的这篇文章主要介绍了ubuntu 安装pytorch前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

firstly github install

then

First make sure your using anaconda'spythonand not the system one.

To installpytorchviacondado

conda install pytorch torchvision -c soumith

or via pip

For python3

pip install http://download.pytorchorg/whlcu75torch-0.1.12.post2-cp35cp35mlinux_x86_64whl pip install torchvision

For python2

cp27none dont install the CUDA version if you dont have Nvidia GPU on your machine that supports.

猜你在找的Ubuntu相关文章