1.在桌面右击,点击显示设置
点击帮助
3.点击系统信息
点击组件
查看cuda版本信息,如图所示我的就是cuda12.1
Window+R 进入cmd,输入nvcc -V查看是否有显示。如果有如图的显示,那么就说明有cuda,如果没有就说明需要下载cuda
- 打开cuda官网,选择对应的版本进行下载。(cuda历史版本:CUDA Toolkit Archive | NVIDIA Developer),图里是我的12.1对应下载的版本。然后点击进去,如果是Win_10就把11换成10。然后点击下载。
双击安装CUDA(路径建议不要更改)
等待下载完成
点击自定义安装
Visual Studio Intergration取消勾选。
等待安装完成即可。
- 安装torch,download.pytorch.org/whl/torch_stable.html这个链接里有各个历史版本。
找到cu开头。选择合适的版本,注意一定要选择cu开头的才是gpu版本
torch后面的数字是torch版本,cp后是对应的python版本,还要注意操作系统
- 下载好后放置在一个文件夹中,如图所示。
在pycharm终端进入到该路径下运行代码:pip install "文件名" 进行安装。
这里我加了清华源,因为我安装时发现我不加清华源会报错。
安装指令为:
pip install torch-2.0.0+cu117-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple
显示Successfully就说明安装成功。
- 检验是否安装成功:
import torch
print("hello:{}".format(torch.__version__))
运行后,系统输出安装版本即代表安装成功。