需求概述
目前在做预测任务,但是cuda算力不足以运行模型,因此想用cpu进行运行,那就需要cpu版本的torch
在之前已经安装过cuda版本的torch
因此自己已经知道我需要安装的torch版本是1.7.0+的
那么我去conda创建虚拟环境,然后在下面网址找到cpu版本的torch轮子,注意自己需要的python版本是什么
https://download.pytorch.org/whl/torch/
接着在创建的虚拟环境中pip install 轮子
就完成了
验证
在命令行输入python(注意一定要在自己创建的环境中)
然后
‘’’
import torch
print(torch.version)
‘’’
文章讲述了如何在CUDA算力不足的情况下,通过在conda虚拟环境中下载并安装CPU版本的PyTorch1.7.0,以及验证安装的步骤,包括导入torch并打印版本信息。
4万+

被折叠的 条评论
为什么被折叠?



