在安装之前，需要先配置GPU环境（安装CUDA和CudaNN)

• 命令行输入nvidia-smi，查看驱动信息

  nvidia-smi

安装相应的CUDA 和CUDANN

验证：输入nvcc --version 或者nvcc -V 进行检查

nvcc --version
nvcc -V

在anaconda里创建环境

conda env list  查看已有环境
conda remove -n  py39gpu  --all   删除已有环境

创建环境 

conda create -n py39gpu python=3.9

激活环境

conda activate py39gpu

 

conda config --set remote_read_timeout_secs 600.0

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge