引入
在4070限购之际,购入3090,之前做了简单的调研,3090和A5000是性价比较高的英伟达GPU.
安装版本
如题示例,window10专业版本+rtx3090+cuda11.8+python10+torch2.0亲测成功
这里主要说明几点:
- 安装cuda前,安装visual studio community2022。
- RTX3090仅支持cuda11以上版本,
- pytorch1.7.0才开始支持cuda11.
- cuDNN根据所选cuda版本安装,此次安装cuDNNv8.9.5
CUDA+cuDNN安装
CUDA安装链接:
https://developer.nvidia.com/cuda-downloads
https://developer.nvidia.com/cuda-toolkit-archive
https://developer.nvidia.com/cuda-11-8-0-download-archive
cuDNN下载链接,需要注册账号登录:
https://developer.nvidia.com/rdp/cudnn-download
python10+pytorch2.0
需要考虑torch对应的安装包问题:
下载网址:
https://pytorch.org/get-started/previous-versions/
安装命令:
pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
参考博客
- https://blog.csdn.net/weixin_70285813/article/details/130925697