win10安装pytorch

在一台新的电脑上配置win10的目标检测环境,由于着急训练,没安装好移动硬盘版linux就先用win10了。

下载anaconda3安装

  1. 创建虚拟环境
  2. 激活虚拟环境
    创建虚拟环境是因为我发现进入虚拟环境后的pip和conda库都是当前虚拟环境下的,能避免环境冲突,待验证。

安装torch

1.第一种去官网直接安装

去torch官网
pip install 巴拉巴拉

#最好用pip,换源方便快捷。
pip install 巴拉巴拉 -i https://pypi.tuna.tsinghua.edu.cn/simple/
#你用conda也要换源,还不如pip

2.因为网络不好,pip下了一半就断开,所以我离线安装了

先去
链接: https://download.pytorch.org/whl/torch_stable.html.

下载对应python版本的和cuda版本的torch和torchvision的.whl文件

查看python版本:
python -V

查看cuda 版本:
nvidia-smi
#就我目前了解,cuda一般现在向下兼容,保持你的cuda版本号高于torch的就行。如果你用别人的代码框架还是按着别人的指定版本来吧。
#12-8更新如果你笔记本驱动仅支持cuda10.2 那么你下载torch-cuda11.0是不行的。以前linux下超了0.几个版本没出问题,可能跨版本不太行,后面再遇到再补充吧。

pip install 巴拉torch巴拉.whl
pip install 巴拉torchvision巴拉.whl

2022.07.24s
工作站上是cuda11.4,我用conda离线安装cuda11.5不行报错,
用pip安装官网离线torch和torchvision 的cuda11.3版本可以了

目前理解是按理说cuda版本要相等,但是实操来看torch和torchvision的cuda版本可以略低于电脑上的cuda版本,但是不能低大版本级别,11.*内应该是可以的。

202207.24e

第三种

----2022.7.12----更新s----
1.下载清华源中的torch和torchvision压缩文件
清华源找自己对应的cuda、python版本的torch和torchvision
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
2.安装
conda install --offline ***.tar.bz2
----2022.7.12----更新e----

测试安好没

python
import torch
torch.cuda.is_available()

输出是true就好了

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值