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
相关推荐
zhizhizhuzhuxia4 分钟前
情感解说短视频的配音选型:从工程视角看情感音色与批量出片
人工智能·音视频
YOLO数据集集合6 分钟前
光伏组件异常检测数据集 | 光伏组件 异常检测 红外检测 热斑识别 光伏巡检9147期
人工智能·yolo·目标检测·语言模型·太阳能板·光伏组件
HIT_Weston23 分钟前
242、【AI】【模型部署】基座模型研究:注意力机制深入
人工智能·模型部署
智碳能碳管理平台24 分钟前
碳排放核算软件选型:历史台账断层补救与锁账回放实操指南
人工智能·数字孪生·能碳管理系统·智碳能碳管理平台·企业能碳管理系统·碳排放核算软件·绿色工厂申报saas
IT_陈寒36 分钟前
SpringBoot自动配置坑了我一把,原来是这样绕过去的
前端·人工智能·后端
hahaha601644 分钟前
Shades-of-Gray (SoG) 算法--白平衡算法
人工智能·嵌入式硬件·算法·计算机视觉
海宇大数据1 小时前
零信任架构实战:基于海宇活体识别V步骤1构建自动化远程公证会话初始化网关
运维·人工智能·架构·自动化
Latchh1 小时前
前端导出的JPEG红字发糊,质量要拉到100才清楚
前端·图像处理·人工智能·计算机视觉
思考着亮1 小时前
2.skill
人工智能
code_slave(码畜)1 小时前
微服务架构落地:基础服务 —— 报表服务(AI 集成篇:AI 增强报表能力)
人工智能·spring boot·spring cloud·微服务·架构