Windows11 GeForce GTX 1060 CUDA+CUDNN+Pytorch 下載及安裝

一、查看顯卡型號信息

系統：Windows11
顯卡：GeForce GTX 1060
型號：
（1）搜索 NVIDIA，選擇 NVIDIA Control Panel

（2）打開 NVIDIA control Panel，打開系統信息，點擊組件，找到 NVCUDA.DLL,查看版本號：11.1.106
說明我們可以下載 11.1.106 及其以下版本的 CUDA
CUDA Driver 版本

二、下載 CUDA

https://developer.nvidia.com/cuda-toolkit-archive
我們選 CUDA Toolkit 11.1.0

選擇 Windows => x86_64 => 10（Windows 10/11 可以用）=> exe (local) => Download
下載 CUDA
下載完成后，點擊安裝

下載 CUDNN

https://developer.nvidia.com/rdp/cudnn-download
選擇 Windows => x86_64 => 11 => exec (local) => Download
CUDNN 下載
下載完成后，點擊安裝

安裝 Pytorch

pip install torch==1.8.0+cu111 \torchvision==0.9.0+cu111 \torchaudio==0.8.0 \-f https://download.pytorch.org/whl/torch_stable.html

測試驗證

# Python 環境
>>> import torch
>>> torch.cuda.is_available()
True

torch.cuda.is_available()

參考

https://blog.csdn.net/weixin_45955630/article/details/113105064

本文來自互聯網用戶投稿，該文觀點僅代表作者本人，不代表本站立場。本站僅提供信息存儲空間服務，不擁有所有權，不承擔相關法律責任。
如若轉載，請注明出處：http://www.pswp.cn/bicheng/94380.shtml
繁體地址，請注明出處：http://hk.pswp.cn/bicheng/94380.shtml
英文地址，請注明出處：http://en.pswp.cn/bicheng/94380.shtml

如若內容造成侵權/違法違規/事實不符，請聯系多彩編程網進行投訴反饋email:809451989@qq.com，一經查實，立即刪除！