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查看使用率
解决办法