conda虚拟环境安装pytorch gpu版

最近创建了新的虚拟环境想要搭建虚拟环境使用MONAI进行医学图像处理。直接创建虚拟环境后monai,显示安装的pytorch是cpu版本。上网查了下,发现需要先安装gpu版本的pytoch。于是又一次开始了pytoch的安装之路。距离上次安装已经过去很久,忘记了具体怎么实现快速安装的,于是上网搜怎么在国内快速安装。下面这个链接上的教程是我用着最直观简单的:
pytorch gpu版安装(普适稳妥,亲测有效)

总结下原博主讲的就是

1.先查看cuda版本,命令行使用nvidia-smi

2.从torch、torchvision、torchaudio下载链接选择和cuda版本号及python版本号对应的torch、torchvision和torchaudio三个组件的whl文件,并下载

3.在虚拟环境中使用pip安装命令依次安装下载的三个whl文件(先安装torch,再安装torchaudio,最后是torchvision)

知乎原作者给的安装命令(只以torch安装示例)是

复制代码
pip install F:\download\pytorchinstall\torch-2.2.0+cu118-cp311-cp311-win_amd64.whl

但是我安装的时候遇到了

复制代码
ERROR: Could not find a version that satisfies the requirement filelock (from torch) (from versions: none)
ERROR: No matching distribution found for filelock

在安装命令后面加了清华源后安装成功

复制代码
pip install F:\download\pytorchinstall\torch-2.2.0+cu118-cp311-cp311-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
相关推荐
狂潮lister几秒前
狂潮 · 元理论v1.0-原创架构设计-AI 网关 | 第9篇:KC-AI‑Gateway(一套可审计的大模型网关完整架构实践)|【完整版】
人工智能·后端
林澈在路上6 分钟前
AI做歌工具怎么拆人声伴奏?分轨导出指南
人工智能·ai音乐·melo音乐
方方洛6 分钟前
ray教程-14-RayTune与超参搜索
人工智能·分布式·机器学习
王吉伟7 分钟前
Anthropic 官方指南:多智能体不等于更强大
人工智能·ai·语言模型·自动化
荆棘鸟智能9 分钟前
扬尘在线监测数据接入与异常清洗:从断点补传到质量审计
人工智能·算法·系统架构
Java内核笔记10 分钟前
同一份知识库,4 种切分 × 3 种召回,纯关键词比纯向量高 13 个点
java·人工智能·spring boot
靠谱的工程师朋友10 分钟前
电子会计档案管理软件科普 核心能力与选型要点解析
网络·数据库·人工智能·电子会计档案
wflynn11 分钟前
Sebastian Raschka 11 小时视频课:用 PyTorch 从零构建推理模型
人工智能·ai
买房动力十足12 分钟前
秒杀商城 + AI 运营:Lua 防超卖、MQ 削峰,AI 负责选品和策划
开发语言·人工智能·lua
forest_0913 分钟前
《大模型原理通俗讲》系列 00|开篇 大模型到底是怎么回事
人工智能·python·prompt·agi