js_reverse/机器学习/01安装pytorch保姆级教程
2023-07-12 23:26:55 +08:00
..
2023-07-12 23:26:55 +08:00
2023-07-12 23:26:55 +08:00
2023-07-12 23:26:55 +08:00
2023-07-12 23:26:55 +08:00

知识点python版本pytorch版本显卡驱动支持的CUDA版本安装Anaconda

安装pytorch最大的难点就是版本兼容问题python版本pytorch版本显卡驱动支持的CUDA版本不管是那个版本对不上都会报错

首先解决显卡问题

我的笔记本显卡是1050Ti显卡驱动版本是397.64怎么才能知道驱动版本是397.64cmd输入nvidia-smi

请求

如果Driver Version小于400先更新显卡驱动

那么这里就需要更新显卡驱动去nvidia官方网站下载

https://www.nvidia.cn/Download/driverResults.aspx/148666/cn/

我这里下载的显卡驱动版本是431.36这里我尝试过500的显卡驱动但发现电脑承受不了散热风扇声音很大。

请求

下载下来exe文件安装双击下一步下一步就OK了

请求

再次cmd输入nvidia-smi查看显卡版本和显卡驱动支持的CUDA版本

请求

到这里显卡驱动问题就搞定了

安装Anaconda

去官方网站下载exe文件这里一定要先清理干净原先的py环境否则可能会出现python环境冲突

请求

https://www.anaconda.com/

这里要注意的是,在安装的时候,勾选创建系统环境变量

进入cmd执行以下代码这里我用的python版本是3.7如果装了python3.10后期就报错

# 先在e盘下创建一个anaconda3_env文件夹
# 创建虚拟环境并安装基础工具
windows: conda create --prefix=e:\anaconda3_env python=3.7
ubuntu: conda create -n 环境名称 python=3.8
# 进入虚拟环境
conda activate e:\anaconda3_env
# 列出虚拟环境
conda env list
# 退出虚拟环境
conda deactivate
# 删除虚拟环境
conda remove --all --prefix "e:\anaconda3_env"

安装pytorch

https://pytorch.org/get-started/locally/

进入官网复制一条安装命令由于官方网站首页给出的安装命令不试用老版本CUDA是11.7而我现在需要的是CUDA Version: 10.1

请求

进入历史版本页面搜索CUDA 10.1

请求

复制命令执行,等待执行结束

这个时候可能出现的报错failed with initial frozen solve. Retrying with flexible solve

请求

更新conda环境

# 查看版本
conda -V
# 更新conda环境
conda update -n base conda
# 更新conda的所有包
conda update -n base -c defaults conda

在更新的过程中报错ImportError: DLL load failed while importing _ctypes

请求

网上找了很多方法最后发现是anaconda2023版本问题我直接降低版本安装

https://repo.anaconda.com/archive/

我测试很多个版本,发现版本之间确实存在兼容问题,选择合适自己本机版本才是上策

做完这些之后再用conda install xxx就不会有上面的问题了。

最后测试

import torch
print(torch.cuda.is_available())

返回 True 就证明安装成功了

这里再次出现报错:找不到指定模块

我这里直接下载Visual Studio Community软件

https://visualstudio.microsoft.com/zh-hans/vs/community/

请求

勾选默认安装就OK了