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.org/whl/torch_stable.html
gpu版下载
conda下载命令
# cuda==10.2
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
# cuda==10.1
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
# cuda==9.2
conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev
pip下载命令
# cuda==10.2
pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html
# cuda==10.1
pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
# cuda==9.2
pip install torch==1.5.0+cu92 torchvision==0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
使用pytorch安装规范操作安装,但是仍然出现导入包的问题。
即在python和ipython中导入出现问题——这里提供No model is "torch"的解决方法之一。
(关于numpy的版本问题,这里没有考虑,因为按照标准步骤安装的,库应该都是一样的,兼容当前版本的torch的)
或许我们是直接进行下载安装的,但是呢?有一个问题可能被忽略了——python版本问题。
版本不兼容,是无法导入的。
如果是使用的Anaconda进行包管理,就只需要执行以下命令即可在当前虚拟环境下自动安装库需要的python版本。
conda install python
如果只是拥有一个python环境(这里指系统下只有一个版本的python且没有Anaconda),那么可能比较麻烦点,需要重新安装对应版本的python。
这里说明一下——关于1.5.0的torch对应的python版本:3.6.6