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。下载特别慢,下载完成后如下图所示:
在这里插入图片描述
用conda安装:conda install pytorch=0.4.0 cuda90 -c pytorch,会安装依赖包,如下所示:
在这里插入图片描述
问题:红框的意思是新安装了cuda吗?
之前命令所选择的cuda版本指的是系统所安装的cuda版本吗?

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

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值