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

 

posted @ 2023-05-25 19:47  壹碗  阅读(38)  评论(0编辑  收藏  举报