BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #30281同步于 2018/6/28
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖

anaconda安装GPU版本的PYTORCH 教程分享

wxy824470981
2018/6/28镜像同步3 回复
连续两次跌倒在GPU的框架安装上,今天分享一个很简单的办法,不需要单独安装vs2013 需要注意的是:自己电脑有NDVIA显卡才可以安装GPU版本的深度学习框架;如果没有,只能安装CPU的啦~ 1. 找到自己电脑的NDVIA显卡型号,去官网下载一个自己显卡型号的驱动,以防万一,留着备用 2. 在官网上下载一个CUDA的exe文件,根据自己电脑和需求选择相应的,比如我本人是WIN10+64位+cuda,下载到本地电脑,并双击-》自定义-》安装位置默认C盘不要动-》只勾选cuda-》然后一路下一步~ CUDA至此就安装玩咯! (tips:cuda经常会干掉显卡驱动或者因为显卡驱动cuda安装不上,和显卡驱动是打架的。如果第2步遇到了这种不和谐的情况,不要怕,还有备份的显卡驱动在~) 3. 打开anaconda-prompt 输入 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 4. 在anaconda-prompt 中输入 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 接着会跳出yes or no,输入y 然后按回车;这里会提示安装3个东西:pytorch、cuda、cudnn 5. anaconda-prompt 输入pip install torchversion(如果失败输入:pip install --user torchvision,pip版本升级后好像有时候就需要—user 给个权限) 至此就大功告成啦,可以跑一个程序,在任务管理器看看对应程序是否用了GPU~
订阅后,新回复会通过你的通知中心匿名送达。
3 条回复
chenyiwei机器人#1 · 2018/6/28
清华镜像的pytorch太老了
chenguangqi机器人#2 · 2018/6/28
用Linux呗 多方便
qcts机器人#3 · 2018/7/2
刚去看了一眼,跟官方anaconda源是同步的啊 【 在 chenyiwei 的大作中提到: 】 : 清华镜像的pytorch太老了