• 博客园logo
  • 会员
  • 众包
  • 新闻
  • 博问
  • 闪存
  • 赞助商
  • HarmonyOS
  • Chat2DB
    • 搜索
      所有博客
    • 搜索
      当前博客
  • 写随笔 我的博客 短消息 简洁模式
    用户头像
    我的博客 我的园子 账号设置 会员中心 简洁模式 ... 退出登录
    注册 登录
村雨sup
自己选的路,跪着也要走完 XD
博客园    首页    新随笔    联系   管理    订阅  订阅
windows下配置cuda9.0和pytorch

今天看了看pytorch官网竟然支持windows了,赶紧搞一个。

下载cuda 9.0  https://developer.nvidia.com/cuda-downloads

下载anaconda3  https://www.anaconda.com/download/

(记得安装在默认位置,不然的话windows下会发生很多错误)

pytorch官网查conda安装pytorch的命令

conda install pytorch -c pytorch
pip3 install torchvision  (或者pip)

在conda中敲入python

>>import torch 

>>torch.cuda.is_available

True

表示安装成功了

感觉比ubuntu简单啊,我的虚拟机可以删了.....

posted on 2018-08-25 13:06  村雨sup  阅读(639)  评论(0)    收藏  举报
刷新页面返回顶部
博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3