2024.04.30更新
安装流程简介
1、安装 Anaconda3
2、检查显卡,更新驱动
3、创建conda虚拟环境
4、在自创的虚拟环境安装 PyTorch(torch版本和torchvision版本一定要对应!!!重点!!!)
5、测试安装是否成功
PS:经过N次的Pytorch安装,每次都要翻大佬们的博客,痛并思痛,还是记录下自己安装历程,以遍日后回顾。
1、安装 Anaconda3
Anaconda 是一个用于科学计算的 Python 发行版,支持 Linux, Mac, Windows, 包含了众多流行的科学计算、数据分析的 Python 包。
Anaconda安装包下载(用清华源下载比较快):
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
2、检查显卡,更新驱动(电脑较新的可以跳过这步)
首先,建议到英伟达NVIDIA官网更新驱动,按照最新的来安装(不更新也行,但更新最新的驱动是为了避免不必要的麻烦):
https://www.nvidia.cn/Download/index.aspx?lang=cn
选择自己的显卡型号,下载安装即可
然后,安装好后进入NVIDIA控制面板(一般电脑右下角就会看到英伟达的图标),鼠标右键可进入。点击帮助,选择系统信息。
点击组件,找到自己CUDA的版本号
也可以 win +R ->cmd ,进入命令行,输入:nvidia-smi ,即可获得cuda版本号
3、 创建conda虚拟环境
找到 Anaconda prompt 命令窗口
创建一个新的虚拟环境
(1)创建命令: conda create -n mypytorch python=3.9
其中,mypytorch是虚拟环境的名字,自己随意设置;python 3.9是指python的版本,一定要指定;
(2)激活环境:conda activate mypytorch
(3)输入以下命令,可查看自己安装的环境有哪些:conda info --envs
4、安装PyTorch(包括torch和torchvision)
PyTorch官网:https://pytorch.org/
不推荐从官网下载安装!!!费劲且慢且问题多!!!
建议使用以下方法:
(1)首先,到这下载 https://download.pytorch.org/whl/torch_stable.html 自己需要的 torch 和 torchvision 版本
其中,cu121就代表cuda12.1,cp310代表python3.10,win代表window系统。
PS:下载的时候,torch和torchvision版本一定要对应!要不然后面会报错。
我自己试出来的版本对应:
torch2.1.0 -> torchvision 0.16.0
torch2.1.1 -> torchvision 0.16.1
下载whl文件之后,激活刚才创建的虚拟环境:conda activate mypytorch
再找到自己刚下载的whl文件存放路径,cd
进入到whl文件的路径。
输入以下命令完成安装:
pip install torch-2.1.1-cp310-cp310-win_amd64.whl
pip install torchvision-0.16.1-cp310-cp310-win_amd64.whl
# 注意,实测有时安装包还是会卡顿,加上清华源的镜像好一点,比如下面这样
pip install torch-2.1.1-cp310-cp310-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/
pip install torchvision-0.16.1-cp310-cp310-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple/
5、测试安装是否成功
打开 Anaconda prompt 命令窗口
进入mytorch虚拟环境:conda activate mytorch
接着,输入python
再输入以下命令:
import torch
torch.cuda.is_available()
显示True,则安装成功。