Windows 下安装 CUDA 和 Pytorch
下载cuda并安装 CUDA Toolkit 11.6 Update 2 Downloads | NVIDIA Developer

查看duda版本

用Anaconda新建一个python3.8的环境
然后去Start Locally | PyTorch官网下载cuda对应的gpu环境(建议使用pip)选择具体细节后,复制命令安装

建议直接复制下面的代码
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio===0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
检测安装

验证
《动手学深度学习》 — 动手学深度学习 2.0.0-beta0 documentation

然后解压


运行jupyter notebook

到达路径


点击

使用nvidia-smi查看使用率


解决办法


浙公网安备 33010602011771号