热门
最新
红包
立Flag
投票
同城
我的
发布
CSDN App 扫码分享
评论
1
打赏
- 复制链接
- 举报
下一条：
安装cpu版torch：pip3 install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html安装gpu版torch：pip3 install torch==1.2.0cu92 torchvision==0.4.0cu92 -f https://download.pytorch.org/whl/torch_stable.html（cu92为对应的cuda版本）查看gpu版本是否安装成功：import torch print(torch.cuda.is_available()) true即成功查看cuda版本：cmd里输入nvcc -V
