刚刚在安装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里面用安装成功
网友评论