Win系统PyTorch+cuda安装+踩坑经验

Win系统下anaconda中PyTorch+cuda安装+踩坑经验

cuda以及cudnn的安装

可以参考以下文章
https://blog.youkuaiyun.com/jhsignal/article/details/111401628
验证安装是否成功
https://blog.youkuaiyun.com/jhsignal/article/details/111398427
想让pytorch成功安装，需要匹配对应的cuda,cudnn版本；安装了这两个就不用额外安装cuda toolkit了,同时注意cuda版本不能超过电脑支持的版本。

//```javascript
//查询电脑支持的最高版本
nvidia-smi;

pytorch的安装

打开anaconda的prompt窗口，创建虚拟环境；
此时窗口应为（base）开头
下面展示一些 内联代码片。

// 输入已下载的python对应版本号
create -n pytorch python==3.11;
//查看虚拟环境是否被创建成功
conda info -e
conda env list
conda activate python;

**此时开头应变成（pytorch）…**不要在base里面下载

到官网查找对应win+cuda版本的pytorch下载命令，此时要用pip下载，千万不可用conda！！用conda会有一定概率下载为CPU版本，但是cuda需要GPU版本才能匹配！！！否则后续步骤torch无法导入成功

把pip命令粘贴到（pytorch）后，回车
等待安装；不成功就多试几次
输入pip list查询，此时torch版本匹配cuda11.8

保持（pytorch）环境，不要退出，输入python，之后输入import torch
等待import完成后，

// 
 print(torch.__version__);
 print(torch.cuda.is_available())

True就代表成功安装了

如果是false，建议pip list查一下torch对应的版本是否是GPU版本
如果不是，先卸载掉torch,再重新装正确版本，如果有多出来的包，也一起卸载掉

//
conda uninstall pytorch

//用conda也行
pip uninstall torch
pip uninstall torchvision

退出pytorch
deactivate

在后续调用jupter notebook时，需要用pytorch模块，要在navigator里选择对应的环境下的jupter notebook打开。

如出现诸如找不到matplotlib模块等问题，在对应的pytorch里下载即可
如：pip install matplotlib;