选择安装版本
INSTALLING PREVIOUS VERSIONS OF PYTORCH
之后执行如下命令安装完整驱动
conda install -c conda-forge cudatoolkit-dev
[CUDA_HOME environment variable is not set. Please set it to your CUDA install root for pytorch cpp extensions](https://github.com/conda/conda/issues/7757)