conda离线安装pytorch的方法

win10系统conda离线安装pytorch的方法,亲测可行

离线安装的原因

首先需要anaconda环境和cuda,可以搜索安装方法。本人工作站比较古老,CUDA版本是10.0,无法安装最新版的pytorch。尝试在线安装pytorch多次失败后,决定离线安装pytorch。首先有很多博客都在使用清华镜像源在线安装pytorch,实际操作中遇到了很多问题,首先清华镜像源可能不存在对应的安装包,其次需要修改镜像源通道和.condarc文件,随后使用官网给出的相关命令安装(去掉-c pytorch),但是还是出现了很多错误,终于决定离线安装。cuda对应版本的torch和torchvision可以参考官网,也可参考博客.

下载对应cuda版本的pytorch和torchvision

推荐使用下载源:https://download.pytorch.org/whl/torch_stable.html.cu开头的是GPU版本的pytorch,注意不要犯低级错误,比如本人下载了linux系统的安装包,结果半天也没安装上。下载好whl文件后,把安装包放到一个文件夹中

安装

管理员方式进入anaconda prompt,conda activate +环境名 激活你要安装pytorch的环境。安装命令为:pip install 路径\文件名 注意文件名要加后缀。安装后输conda list 查看有没有安装好。

测试

conda激活环境,进入python,输入以下代码:

import torch
import torchvision

如果没有报错,证明安装成功,接下来检测GPU是否成功调用:
conda 安装环境下输入:

print(torch.cuda.is_available())

输出true证明成功
或者pycharm中运行:

import torch

print(torch.version.cuda)  # 查看 CUDA 版本
print(torch.cuda.is_available())  # 查看 CUDA 是否可用(即训练时是否可用 GPUprint(torch.cuda.device_count())  # 查看可行的 CUDA 数目

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3, 3).cuda())

可以成功输出就大功告成了。

  • 3
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值