Ubuntu16.04配置Pytorch记录

显卡: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

猜你喜欢

转载自blog.csdn.net/Mr_Pingg/article/details/123382645
今日推荐