1.首先进入pytorch官网:pytorch
-
根据自己的版本和配置来安装,我系统是linux,我采用的pip来安装,python版本是3.7,cuda版本是10.1。
2.输入命令行 : -
原命令行是:pip3 install torch torchvision
我修改后的命令行为再后面添加了:
-i https://pypi.tuna.tsinghua.edu.cn/simple/, -
这样做的目的是加速pip的下载速度,pip 之所以慢,是因为它默认使用的国外的统一的下载源,所以要想加快pip的下载速度,可行的办法就是切换下载源为国内,国内一些大公司也提供了对应的竞相网站。
-
所以完整的命令行为:
pip3 install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple/
- 可是安装好以后发现进入python, import torch 出错,no module named torch.
3.经网上查找资料发现,原因在於pip和pip3會將軟件包安裝在不同的目錄下,如圖:
- 可见,若启动了anaconda的话,应该使用pip而不是pip3来安装软件包,因为只有pip才能把软件安装在与anaconda兼容的目录下。
4.重新进入python,导入torch
5.成功!!!!!