之前安装pytorch非常慢:
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
但是以下方法中下载安装pytorch真的是非常快(在本机安装):
pip3 install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
因为是在conda创建的虚拟环境中安装,所以需要使用以下命令:
pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
安装完成后可以使用pip list 查看安装的包是否存在,注意不能使用pip3 list,这条命令执行出现的包不是虚拟环境中的
存在的问题,使用ipython测试import torch时,会报错,但实际是因为直接输入ipython,但是虚拟环境中没有安装,会直接调用虚拟环境外的ipython,虚拟环境外的ipython中不含torch,所以会报错。