1:打开CMD nvidia-smi.exe 查询显卡
cuda VERSION:12.3
2:打开https://pytorch.org/get-started
我的系统是12.3
3:使用google 搜索 nvidia developer
NVIDIA DeveloperCUDA Toolkit - Free Tools and Training | NVIDIA Developer
目前可以下载:12.3版本,但是pytroch目前仅支持12.1版本
我们需要下载历史版本
因为目前仅仅支持12.1版本。我们需要下载12.1
配置下载选项
安装完毕之后,在CMD控制台输入:nvcc -v
如果显示:代表安装成功
如何卸载CUDA呢?
控制面板找到 nvdia CUDA 全部卸载
4:pytroch相关依赖库安装
https://pytorch.org/get-started
在之前的环境中,我们安装了cuda 12.1,所以这次我们在这里选择12.1
打开CMD输入(下列代表,在上图 RUN THIS COMMAND 已标记)
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
我的电脑环境目录是:C:\Users\Administrator\AppData\Local\Programs\Python\Python310\Scripts
安装完毕后输入:
python
import torch
print(torch.cuda.is_available())
出现false失败 出现true成功