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

2020-07-02  本文已影响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里面用安装成功

上一篇下一篇

猜你喜欢

热点阅读