ubuntu 安装pytorch

来源:互联网 发布:周扬青淘宝店铺货源 编辑:程序博客网 时间:2024/05/21 04:44

firstly github install 

then

First make sure your using anaconda's python and not the system one.

To install pytorch via conda do

conda install pytorch torchvision -c soumith

or via pip

For python3

pip install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl pip install torchvision

For python2

pip install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp27-none-linux_x86_64.whl pip install torchvision

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


原创粉丝点击