Pytorch和CUDA10.2环境配置

╰+哭是因爲堅強的太久メ 2022-08-29 06:13 218阅读 0赞

近期开始学习pytorch，又是一顿配置环境的瞎折腾，我电脑上的CUDA原来版本是9.x的，现在好多新算法的要求是10.0以上，就覆盖安装了10.2版本的CUDA。

> 参考资料：  
> [windows系统查看CUDA版本号][windows_CUDA]  
> [深度学习环境搭建（GPU）CUDA安装（完全版）][GPU_CUDA]  
> [tensorflow各个版本的CUDA以及Cudnn版本对应关系][tensorflow_CUDA_Cudnn]  
> CUDA10.2安装文件：[提取码：daks][daks]

检查安装CUDA10.2是否正确`nvcc -V`：  
![在这里插入图片描述][f28cd0c751094d11948ef3cf0fab1ad5.png]  
显示CUDA信息则安装成功，然而我在测试cuda的gpu是否可用的时候，却显示错误：

import torch
    
    print(torch.__version__)	#1.6.0
    print('gpu:', torch.cuda.is_available())	# False

后来想明白可能是CUDA和torch版本不兼容，卸载现有torch，去[PyTorch官网][PyTorch]找到相应版本进行下载测试：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2tlaXZlbl8_size_16_color_FFFFFF_t_70]  
显示成功！！！  
![在这里插入图片描述][6df8e6f2fa674e08b468c47c5cf93316.png]

[windows_CUDA]: https://cloud.tencent.com/developer/article/1351824
[GPU_CUDA]: https://blog.csdn.net/cashmood/article/details/105081586
[tensorflow_CUDA_Cudnn]: https://blog.csdn.net/qq_27825451/article/details/89082978
[daks]: https://pan.baidu.com/s/12sXuJNPz9_KaGSPSHt_u6Q
[f28cd0c751094d11948ef3cf0fab1ad5.png]: /images/20220829/373ec778d1c140b8a13f34e1c90eca83.png
[PyTorch]: https://pytorch.org/get-started/locally/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2tlaXZlbl8_size_16_color_FFFFFF_t_70]: /images/20220829/2810801dfb8e44b6abc35f4d04320e36.png
[6df8e6f2fa674e08b468c47c5cf93316.png]: /images/20220829/b0e9fbfa3f1045f2a571891003a6fdac.png