1、不改清华源的前提下,官网下载太慢,如果换清华源又不能兼顾CUDA,最简单的办法直接多线程下载。小白可以用迅雷下载。
2、cuda版本和显卡支持版本需要自己看一下。
安装环境:window10
显卡:3050ti
使用Anaconda3-2024.06-1-Windows-x86_64
内置python版本是:python3.12.4
Pytorch版本:2.4.1+cu124
CUDA版本:12.4
1、安装版本及查看
没显卡的可以直接在Anaconda中的Anaconda Prompt
在命令提示行中直接输入
python
pip install pytorch -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
有显卡打开NVIDIA控制面板。
1、打开NVIDIA控制面板。
2、点击系统信息。
3、点击组件。查看NVIDIA CUDA 版本。
如图:我安装的是 12.4
这里建议的CUDA版本是11.8、12.1、12.4
如果CUDA版本是11.8-12.4范围内的可以跳过CUDA更新及安装。
建议去torch查看支持范围:Pytorch官网
确定好需要安装的CUDA版本后,访问NVIDIA的CUDA下载界面
安装CUDA方法:点击进入 CUDA-toolkit-archive
选择合适的版本进行安装。
3、安装Pytorch
这个是截止到2024年9月,Pytorch的安装环境需求。
如果CUDA安装的版本是12.4。可以使用下面这个命令下载
python
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
下载太慢,直接复制链接通过迅雷下载
迅雷下载后,打开文件夹位置
在文件导航栏中输入CMD
,
点击回车并输入以下命令:
python
pip install
接着将下载的Pytorch拉倒CMD
中
点击回车运行。等待安装成功。
python
python
import torch
torch.cuda.is_available()
返回True就代表可以安装完成了。