pytorch安装
电脑配置:
window10、 python3.6 cuda11.4.14X
一、CUDA安装
(1)查看电脑版本
(2)下载cuda,大约2.8g。网址(CUDA Toolkit 11.4 Update 1 Downloads | NVIDIA Developer)
(3)安装cuda,默认安装路径C:\Users\39211\AppData\Local\Temp\CUDA
(4)检验:nvcc -V
二、pytorch安装
pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f pip3 install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html