搞了几天终于安装下来了
1.配置conda
2.(电脑用的nvidia显卡的话)下载cuda,没有则不下
3.配置cuda环境变量
环境变量.png
4.通过pip安装与cuda对应版本的pytorch
(用conda下太慢了,pip好一点不过也容易断,重下了好几次)
5.将pytorch配置到pycharm上:
setting - python interpreter - 右上角设置图标add - Virtual Environment - Exist Environment - 选择之前安装pytorch的文件夹里的python环境 - 然后就搞定了,等待加载完即可。
python版本:3.7
cuda版本:11.1
pytorch版本:1.8
nvidia版本:gtx1050
吐槽1:cuda会自动安进c盘,安装时设置的路径都只是临时的
吐槽2:看很多攻略说要安装cudnn,我也跟着安装了,但测试cudnn的时候也没测试成功,就直接跳过了,直接用pip去下pytorch,也能import torch,就是不知道cudnn起没起作用。
吐槽3:pycharm配置pytorch的时候弄了好久,电脑里的python太多了,一定要选安装pytorch时用的python。
网友评论