安装显卡驱动
sudo ./NVIDIA-Linux-x86_64-450.66.run -no-x-check -no-opengl-files -no-nouveau-check
Gitlab
目前只能使用cuda10.1
pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
https://pytorch.org/get-started/previous-versions/