Win10安装torch和torchvision的CPU版本

论坛 期权论坛 编程之家     
选择匿名的用户   2021-6-2 19:10   2384   0

Windows10+Anaconda3的环境。

1)启动Anaconda Powershell Prompt (Anaconda3)的环境;

2)创建CpuTorch的环境,conda create -n CpuTorch python=3.6

3)完成后进行激活, conda activate CpuTorch

4)开始安装torch和torchvision的CPU版本;

访问:https://pytorch.org/get-started/locally/

安装命令为:【图在后面。】

pip install torch==1.8.0+cpu torchvision==0.9.0+cpu torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

5)安装完成后,输入python,再输入import torch

import torchvision,没有报错就OK了。

检查torch和torchvision的版本信息

print(torch.__version__)和print(torchvision.__version__)

分别显示:1.8.0+cpu和0.9.0+cpu

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP