admin管理员组

文章数量:1530518

安装pytorch出现程序无法定位入口(import torch 报错)的问题解决方案

  • 问题描述
  • 问题原因
  • 解决方法

                         QQ:3020889729                                                                                 小蔡

问题描述

pytorch安装方法如下:(简单选择系统和下载方式即可得到需要的下载命令)


这里把命令放一下,需要的可以直接使用
cpu版的torch下载

# cpu版  ——也就是None版
# conda下载方式
conda install pytorch torchvision cpuonly -c pytorch
# pip下载方式(较慢——因为后一个下载点是外网)——你可以分开自己单独下载,不过我还是推荐使用conda,方便快捷
pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch/whl/torch_stable.html

gpu版下载
conda下载命令

# cuda==10.2
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
# cuda==10.1

本文标签: 报错入口解决方案程序Pytorch