在安装好CUDA和CuDNN的前提下。安装步骤见Anaconda配置一套带有tensorflow-gpu新环境的顺序
去PyTorch官网
点击GetStarted 或者 install
生成安装命令(注意CUDA版本要和你安装的对应,如果没有对应的,安装相邻的)
如:
pip3 install torch1.9.0+cu111 torchvision0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
然后到anaconda的环境中执行