在Windows中安装PyTorch及Torchvision

1. 创建虚拟环境

具体查看我之前写的
《在Windows中利用Python的venv和virtualenv创建虚拟环境》

2. 检查显卡版本和CUDA

这种情况是需要电脑上有单独的英伟达的显卡、或者英伟达的显卡和集显这两种情况都是可以的。)

在这里插入图片描述

3. 下载链接

进入PyTorch官网选择合适自己设备的下载链接
在这里插入图片描述

4. 下载

进入cmd 输入上一步获得的下载链接
在这里插入图片描述

5. 等待

在这里插入图片描述

6. 检测

下载完成后,进入python环境检测torch是否可用

在这里插入图片描述

7.下载轮文件

轮文件官网
按CTRL+F 搜索对应版本的文件
使用Motrix进行下载轮文件(只是因为torch太大采用下载器下载,其他可直接使用浏览器下载),版本对应如下

在这里插入图片描述

8.下载Torchvision

github pytorch和torchvision版本对应官网
在这里插入图片描述
我的对应版本为
在这里插入图片描述

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值