Linux下创建名字为pytorch1.2.0以及python为3.6.0的虚拟环境:
conda create -n pytorch1.2.0 python=3.6.0
激活bash环境:
source activate
激活虚拟环境:
conda activate pytorch1.2.0
安装pytorch1.2.0:
conda install pytorch==1.2.0 torchvision==0.4.0
查看安装是否成功,输入:
conda list
可以看到pytorch1.2.0安装成功,为了保险起见,我换种方法验证pytorch是否可用,输入
python
接着输入
import torch
出现ImportError: /root/miniconda3/envs/pytorch1.2.0/lib/python3.6/site-packages/torch/lib/libtorch_python的报错,看了网上的解决方案,是因为python版本与pytorch版本不对应导致的,升级python版本即可,输入
conda install python=3.6.2
python的版本会自动从3.6.0升级到3.6.2
再次输入
python
import torch
print(torch.__version__)
出现了正确界面:
顺便可以查看cuda是否可用,输入:
print(torch.cuda.is_available())
如果为True,表示可用;为False,则不可用。