Win10机器学习环境搭建—pycharm、anaconda、pytorch

安装版本说明:
        Python 3.7
        pycharm 2020.1
        anaconda 4.5.11
        CUDA 11.3
        cudnn 8.2.1

1、安装 pycharm

【pycharm安装教程】pycharm安装教程 

【pycharm创建项目教程】Pycharm创建项目教程

2、安装 anaconda

anaconda 是做什么的?
https://www.zhihu.com/question/317083663

【anaconda安装教程】
https://blog.csdn.net/weixin_43682519/article/details/109326081

anaconda官方下载地址:
https://repo.anaconda.com/archive/

anaconda镜像下载地址:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/?C=N&O=D

【另一个anaconda安装教程】
https://www.cnblogs.com/IT-LearnHall/p/9486029.html

安装好anaconda提示:'conda' 不是内部或外部命令,也不是可运行的程序或批处理文件

这是因为环境变量没有添加或者少添加造成的。

解决办法:
https://blog.csdn.net/mars_xiaolei/article/details/82798640

添加好环境变量后测试,可以看到conda版本4.5.11
Win10机器学习环境搭建—pycharm、anaconda、pytorch_第1张图片

3、Anaconda虚拟环境中安装CUDA

如何查看cuda的版本?查看CUDA版本

【Anaconda创建虚拟环境教程】
https://blog.csdn.net/guiddok/article/details/117514187

【Anaconda虚拟环境中安装CUDA教程】
https://blog.csdn.net/Sir666888/article/details/122073675

我安装的CUDA 11.3.0 自动安装对应版本cudnn 8.2.1

 4、用anaconda安装pytorch

代码调试遇到问题:No module named ‘torch‘

需要先安装pytorch,然后再导入torch包

Win10机器学习环境搭建—pycharm、anaconda、pytorch_第2张图片

PyTorch官网:PyTorch

大佬的安装教程,太棒了!!!给大佬三连+鸡腿 

https://b23.tv/UswNcwR
视频里面介绍了anaconda下从创建虚拟环境,一步步如何安装pytorch,最后在pycharm工程中如何添加虚拟环境。

【Pytorch下载网速慢的解决办法】
https://blog.csdn.net/opencv_fjc/article/details/106207611

注意!!!坑出现了!!!,官网给出的安装命令中,-C参数指明了Pytorch安装的下载源,在这里执行该命令的时候,一定要把-C参数连带后面的pytorch去掉,否则龟速会再次降临!

【更换Anaconda的下载源为国内源】
https://blog.csdn.net/opencv_fjc/article/details/106186785

5、CUDA driver initialization failed

安装好pytorch后问题:CUDA driver initialization failed, you might not have a CUDA gpu.Win10机器学习环境搭建—pycharm、anaconda、pytorch_第3张图片

这里是因为CUDA需要对应合适版本的显卡驱动。

我的CUDA=11.3,那么我的显卡驱动版本就要>=452.39Win10机器学习环境搭建—pycharm、anaconda、pytorch_第4张图片

查看显卡驱动版本方案一:在cmd下使用nvidia-smi命令,如图Win10机器学习环境搭建—pycharm、anaconda、pytorch_第5张图片

Windows NVIDIA Corporation下没有NVSMI文件夹解决方法:
https://blog.csdn.net/Kelly_Young/article/details/96848042

 查看显卡驱动版本方案二:直接打开NVIDIA控制面板查看Win10机器学习环境搭建—pycharm、anaconda、pytorch_第6张图片

查看到我的显卡驱动版本过低,这时需要升级驱动

【window10系统英伟达NVIDIA显卡驱动和CUDA软件的安装和升级】
https://blog.csdn.net/qq_36744449/article/details/117913661

成功升级驱动到457.51版本 ,如下图所示
Win10机器学习环境搭建—pycharm、anaconda、pytorch_第7张图片

然后在安装CUDA的虚拟环境下验证CUDA是否可以正常使用

print(torch.cuda.is_available) 

Win10机器学习环境搭建—pycharm、anaconda、pytorch_第8张图片

输出True

大功告成!

by chaikeya 2022/4/19

你可能感兴趣的