美文网首页
2020-07-02-《pytorch学习--001》

2020-07-02-《pytorch学习--001》

作者: 月半弯中影 | 来源:发表于2020-07-02 15:01 被阅读0次

    刚刚在安装pytorch,自己有几点体会:

    1、显卡、显卡驱动、cuda最好匹配。

    2、conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev,用这个命令行在anaconda里面用安装失败

    pip install torch==1.5.1+cu92 torchvision==0.6.1+cu92 -f https://download.pytorch.org/whl/torch_stable.html,用这个命令行在anaconda里面用安装成功

    相关文章

      网友评论

          本文标题:2020-07-02-《pytorch学习--001》

          本文链接:https://www.haomeiwen.com/subject/sslrqktx.html