pytorch学习01-CUDA&CUDNN pytorch安装记录

系统版本

鼠标右键打开NAVIA控制面板查看信息 CUDA:11.6.127

TensorFlow、PyTorch各版本对应的CUDA、cuDNN关系

https://blog.csdn.net/caiguanhong/article/details/112184290

pytorch下载

https://pytorch.org/get-started/previous-versions/
#CUDA 10.2
conda install pytorch1.10.1 torchvision0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch

安装包单独下载:
https://download.pytorch.org/whl/cu102(cu+cuda版本号)

安装cuda

https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
10.2

对应cudnn

https://developer.nvidia.com/rdp/cudnn-archive
Download cuDNN v8.1.0 (January 26th, 2021), for CUDA 10.2

你可能感兴趣的:(pytorch,python,pytorch)