Morecoin Nodepad 全局重载运算符 Spring Cloud Kotlin SCI 动态条形图 regex triggers jquery遍历元素 iot系统 爬虫数据清洗 图片生成链接 kubernetes集群 python零基础教程 java中的string java数组排序 java连接sql pascal教程 windows7loader vbscript程序员参考手册 电脑必备软件排行榜 数据挖掘原理与算法 listpreference 如何给黑白照片上色 html5下载 流程图工具 华为ff hyqihei layout下载 c程序 grep正则表达式 掌门一对一官网下载 3dmax布尔运算 打开组策略的命令 声如银铃 tor下载 flash制作教程 自动聊天机器人 nsa文件
当前位置: 首页 > 学习教程  > python

在 window 上安装GPU版本的torch

2021/2/7 9:55:42 文章标签: 测试文章如有侵权请发送至邮箱809451989@qq.com投诉后文章立即删除

pip install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp37-cp37m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple some-package如果你安装完毕后出现了如下错误: Could not install packages due to an Environme则你只需要在Terminal中 再…

pip install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp37-cp37m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

如果你安装完毕后出现了如下错误:

 Could not install packages due to an Environme

则你只需要在Terminal中
再执行一遍以下代码就行了:

pip install https://download.pytorch.org/whl/cu80/torch-1.0.0-cp37-cp37m-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

至于torchvision, 虽然我早就安装过了,但仍需重新安装安装。因为,在此处也只是用新下载的GPU1.0.0版本的torch把原来的CPU1.6.0版本的torch替换了而已,那么torchvision也要使用对应的版本。所以在Terminal里执行如下命令:

pip install --no-deps torchvision==0.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

然后验证一下:

import torch

print(torch.cuda.is_available())
print(torch.__version__)

输出:

True
1.0.0

由此大功告成!


本文链接: http://www.dtmao.cc/news_show_2000216.shtml

附件下载

相关教程

    暂无相关的数据...

共有条评论 网友评论

验证码: 看不清楚?