成功步骤:
心路历程:曾尝试过conda,pip,pip3(安装成cpu的救命),whl安装都有问题
1.安装anaconda(一定要和你的Python版本匹配!!!)
配环境变量
创建虚拟环境
激活虚拟环境
conda activate NewEnv(anaconda prompt中)
2.安装cuda11.8(还没配置环境和path变量)
参考
Win10安装anaconda 和cuda+cudnn详细教程_anaconda是从哪载入cudnn.ddl文件的-CSDN博客
我的CUDA安装路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
3.安装pytorch
参考
直接在虚拟环境NewEnv中输入:
pip --default-timeout=1000 install torch==2.0.0+cu118 torchvision==0.15.0+cu118 torchaudio==2.0.0 -f https://download.pytorch.org/whl/torch_stable.html
4.pycharm中conda环境的配置
参考
超详细的Pycharm(2023)使用conda配置教程_pycharm配置conda环境_sweet_sunflower的博客-CSDN博客
下载CV库:
pip install C:\Users\Lenovo\Downloads\opencv_python-4.5.5-cp39-cp39-win_amd64.whl
我的情况:
「显卡驱动版本」:526.56
「 CUDA 版本」:
cuda驱动版本:12.0
一般来说都是为了安装CUDA才会来确定驱动版本。这里也会出现CUDA Version:11.4 这里指的是,电脑可以安装最高版本是11.4,我们可以安装低版本的CUDA的。
pytorch,torchvision与python版本对应关系及安装命令:
https://blog.csdn.net/WOSHIRENXIN/article/details/127415609?ops_request_misc=&request_id=&biz_id=102&utm_term=%E7%94%A8oncda.pip.pip3.whl%E5%AE%89%E8%A3%85pytorch%E7%9A%84%E5%8C%BA&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduweb~default-4-127415609.142^v96^pc_search_result_base2&spm=1018.2226.3001.4187
疑难杂症:
1.nvcc和nvidia-smi显示的版本不一致?
【CUDA】nvcc和nvidia-smi显示的版本不一致? - 简书
如果你的CUDA均通过CUDA Tooklkit Installer来安装,那么runtime api 和 driver api的版本应该是一致的,也就是说, nvcc --version 和 nvidia-smi 显示的版本应该一样。否则,你可能使用了单独的GPU driver installer来安装GPU dirver,这样就会导致 nvidia-smi 和 nvcc --version 显示的版本不一致了。
快捷键:
打开设备管理器:win+x