python3.8_CUDA安装

1.在桌面右击,点击显示设置

点击帮助

3.点击系统信息

 点击组件

 查看cuda版本信息,如图所示我的就是cuda12.1

Window+R 进入cmd,输入nvcc -V查看是否有显示。如果有如图的显示,那么就说明有cuda,如果没有就说明需要下载cuda

  1. 打开cuda官网,选择对应的版本进行下载。(cuda历史版本:CUDA Toolkit Archive | NVIDIA Developer),图里是我的12.1对应下载的版本。然后点击进去,如果是Win_10就把11换成10。然后点击下载。

双击安装CUDA(路径建议不要更改)

 

等待下载完成

点击自定义安装

Visual Studio Intergration取消勾选。

等待安装完成即可。

  1. 安装torch,download.pytorch.org/whl/torch_stable.html这个链接里有各个历史版本。

找到cu开头。选择合适的版本,注意一定要选择cu开头的才是gpu版本

 torch后面的数字是torch版本,cp后是对应的python版本,还要注意操作系统

  1. 下载好后放置在一个文件夹中,如图所示

在pycharm终端进入到该路径下运行代码:pip install "文件名" 进行安装

这里我加了清华源,因为我安装时发现我不加清华源会报错。

   安装指令为:

pip install torch-2.0.0+cu117-cp38-cp38-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple

 显示Successfully就说明安装成功。

  1. 检验是否安装成功:

import torch

print("hello:{}".format(torch.__version__))

运行后,系统输出安装版本即代表安装成功。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值