pytorch安装

简直了简直了,又装了一天pytorch,不就是想把代码运行起来吗?配环境咋那么困难呢?
现在的问题是,之前运行ga-rpn代码时,好像是用pip install命令安装了对应CUDA9.0的pytorch1.1。那么现在想要安装多个版本的pytorch,我该怎么办呢?
1.用conda创建虚拟环境
2.激活虚拟环境
3.安装pytorch对应版本。那么用pip install安装呢?还是用conda install安装呢?
用pip安装,不会安装依赖包:pip install torch==0.3.1 -f https://download.pytorch.org/whl/cu90/stable。下载特别慢,下载完成后如下图所示:
pytorch安装_第1张图片
用conda安装:conda install pytorch=0.4.0 cuda90 -c pytorch,会安装依赖包,如下所示:
pytorch安装_第2张图片
问题:红框的意思是新安装了cuda吗?
之前命令所选择的cuda版本指的是系统所安装的cuda版本吗?

anyway,现在算是解决了。解决方法:激活虚拟环境,在虚拟环境下进入python,输入import torch,不报错,输入torch.version,输出对应pytorch版本,输入torch.cuda.is_available(),输出True。证明当前环境下已经安装了想要的pytorch版本,并且可以使用cuda。

你可能感兴趣的:(基础环境)