需求概述
目前在做预测任务,但是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)
‘’’