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
相关推荐
来两个炸鸡腿11 小时前
【Datawhale2604】Hello-agents task01 智能体经典范式构建
人工智能·大模型·智能体
njsgcs11 小时前
我需要ai理解鼠标在工程图里的位置,要能理解标注的任务
人工智能
AI大法师11 小时前
从 Firefox Kit 看懂品牌升级的正确顺序
大数据·人工智能·设计模式·firefox
程砚成11 小时前
小微美业的生存突围:当 “小而美” 遇上轻量化数字化
人工智能
杜子不疼.11 小时前
Python多模态AI开发指南:让AI同时理解文字、图片和语音
开发语言·人工智能·python
IT_陈寒11 小时前
SpringBoot自动配置的坑把我埋了半小时
前端·人工智能·后端
nap-joker11 小时前
利用多模态神经影像数据进行阿尔茨海默病分类和检测的可解释性尖峰神经网络框架
人工智能·深度学习·神经网络
紧固视界11 小时前
设备防松如何选用垫圈和挡圈?常见方案与应用解析_6月上海紧固件展
大数据·人工智能·紧固件·上海紧固件展·紧固件展·上海紧固件专业展
小饕11 小时前
RAG学习之 - 检索质量评估指标详解:从概念到代码实战
开发语言·人工智能·python·学习
搬砖者(视觉算法工程师)11 小时前
关于英伟达的世界模型Cosmos的介绍
人工智能