显卡:NVIDAI 1050
显卡驱动:ubuntu-drivers devices推荐的的驱动384.111
选择Anaconda下配置Pytorch
版本:Anaconda5.2.0+python3.6.5+cuda9.0+cudnn7.05+pytorch(1.1.0/1.0.1/1.0.0/0.4.1_选第一个)
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch
nvidia 440.44+cuda10+anaconda5.2.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch