Windows 11+RTX3060安装CUDA、cuDNN与Pytorch的正确版本 |
您所在的位置:网站首页 › 七彩虹rtx3060各个版本 › Windows 11+RTX3060安装CUDA、cuDNN与Pytorch的正确版本 |
实验环境为RTX 3060的Windows 11。我在一开始没有安装合适的CUDA版本,导致最终Pytorch训练时报错,最终只能推倒重来,浪费了不少时间。 为了提醒更多的同学在一开始就安装正确的版本,在此记录。 经实验,完全没问题的版本是: 1. CUDA Toolkit 11.1.1 CUDA Toolkit 11.1 Update 1 Downloads | NVIDIA DeveloperPlease Note: We advise customers updating to Linux Kernel 5.9+ to use the latest NVIDIA Linux GPU driver R455 that will be available for download from NVIDIA website and repositories, starting today. Select Target Platform Click on the green buttons that describe your target platform. Only supported platforms will be shown. By downloading and using the software, you agree to 2. cuDNN v8.1.1 (Feburary 26th, 2021), for CUDA 11.0,11.1 and 11.2 cuDNN Archive | NVIDIA Developer 安装指令为: conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge如果这篇博客对你有帮助,请给博主点个赞鼓励一下吧! |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |