记录一下,conda可以直接安装cudatoolkit

conda install cudatoolkit=10.1

original refer