• 如果您觉得本站非常有看点,那么赶紧使用Ctrl+D 收藏吧
  • 技术支持 老斯机网店:bbezu.taobao.com/

Pytorch常用安装命令集合

深度学习 laosiji 3857次浏览 0个评论

基于anaconda

基于python2.7,3.5,3.6,3.7     cuda10.1下 conda 安装命令(ubuntu和windows一样命令)

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
 conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

基于python2.7,3.5,3.6,3.7    cuda9.2下 conda 安装命令(ubuntu和windows一样命令)

conda install pytorch torchvision cudatoolkit=9.2 -c pytorch

 

pip安装方式 python2.7  cuda9.2

pip install torch==1.3.1+cu92 torchvision==0.4.2+cu92 -f https://download.pytorch.org/whl/torch_stable.html

python3.5,3.6,3.7 cuda9.2

pip3 install torch==1.3.1+cu92 torchvision==0.4.2+cu92 -f https://download.pytorch.org/whl/torch_stable.html

基于python2.7,3.5,3.6,3.7     cuda10.0下 conda 安装命令(ubuntu和windows一样命令)

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

 

python3.6 cuda10  仅linux

pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl

pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp36-cp36m-linux_x86_64.whl

python3.7 cuda10 仅linux

pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-linux_x86_64.whl

pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-linux_x86_64.whl

python3.7 cuda10.1 仅linux

pip install torch==1.7.0+cu101 torchvision==0.8.1+cu101 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

python3.7 cuda10.2 仅linux

pip install torch torchvision

python3.7 cuda11.0 仅linux

 pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

python3.7 cuda10 windows

pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-win_amd64.whl

pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-win_amd64.whl

基于python2.7,3.5,3.6,3.7     cuda9.0下 conda 安装命令(ubuntu和windows一样命令)

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

python3.7 cuda9 windows

pip install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp37-cp37m-win_amd64.whl

pip install https://download.pytorch.org/whl/cu90/torchvision-0.3.0-cp37-cp37m-win_amd64.whl

基于python2.7,3.5,3.6,3.7     cuda9.1下 conda 安装命令(ubuntu和windows一样命令)

conda install pytorch cuda91 -c pytorch

Python3.6+pip安装gpu版本
目前gpu版本支持cuda8.0,cuda9.0和cuda9.1,请选择对应的版本下载安装,不要同时执行下面三个命令!

pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl
pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl
pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl

Python3.5+pip安装gpu版本
目前gpu版本支持cuda8.0,cuda9.0和cuda9.1,请选择对应的版本下载安装,不要同时执行下面三个命令!

pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl
pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-win_amd64.whl
pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp35-cp35m-win_amd64.whl

Python3.6+pip安装cpu版本

pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl

Python3.5+pip安装cpu版本

pip install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl

conda 安装 cpu版本 所有python版本通用 (windows和linux)

conda install pytorch-cpu -c pytorch

喜欢 (14)
0 0 投票数
文章评分
订阅评论
提醒
guest
0 评论
内联反馈
查看所有评论
0
希望看到您的想法,请发表评论。x
()
x