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查看使用率

解决办法 

 

posted @ 2022-05-15 21:48  贪睡地蜗牛  阅读(166)  评论(0编辑  收藏  举报