pytorch安装

一、安装显卡驱动最新版本

1、nvidia显卡驱动更新链接

官方驱动 | NVIDIAhttps://www.nvidia.cn/Download/index.aspx?lang=cn

2、查看显卡型号

打开任务管理器，找到nvidia的型号，这里是NVIDIA GeForce MX150

 3、选择要下载的驱动类型

对照自己电脑的显卡类型选择驱动类型

下载类型默认即可

4、下载安装

安装路径可以任选，其他全部默认即可

二、确定cuda driver的版本

打开命令窗口，Windows+r，输入nvidia-smi，查看cuda driver版本号，这里是12.2

三、确定cuda runtime版本，并下载pytorch

 1、打开pytorch官网

PyTorchhttps://pytorch.org/

2、选择版本

选择比12.2小的版本即可，这里选择cuda 11.8

 如果需要更小版本而选项中没有提供，可以下拉点击previous... 查看更早版本

另外ctrl+f可以实现浏览器页面的文字搜索，很好用的功能，可以帮助我们实现对所需版本的查找。

如果安装了anaconda即选择conda包，也可选择pip，但conda优先。

3、创建一个虚拟环境，存放pytorch

创建虚拟环境：conda create -n 环境名称 python=3.11

4、复制命令进行下载（cudnn不需要单独安装，在pytorch中有集成）

如果下载过慢，建议使用镜像，使用镜像需要选用pip，复制改代码，并加上一行：

-i https://pypi.tuna.tsinghua.edu.cn/simple some-package

这样是错的！！！镜像只能下载cpu的torch！！

参考下面这个博客(10条消息) torch.cuda.is_available()返回false——解决办法_Nefu_lyh的博客-CSDN博客https://blog.csdn.net/qq_46126258/article/details/112708781（1）删除原来的cpu版本的torch、torchvision、torchaudio，地址如下：

 删除内容如下:

 (2)直接在网站中下载对应的torch、torchvision、torchaudio版本

网址：

download.pytorch.org/whl/torch_stable.htmlhttps://download.pytorch.org/whl/torch_stable.html注意win_amd64等于win x86_64

(3)用cmd打开下载的路径，本别执行：

pip install +名字，先下载torch那个，名字要加后缀

四、检测安装结果

依次输入这三句

python
import torch
torch.__version__
torch.cuda.is_available()

输出torch版本号，证明安装好了

如果输出TURE就证明GPU可以用了

当当！！