Windows下安装CUDA和Pytorch

目录

1.cuda toolkit Download

2.cuDNN Download

3.pytorch下载安装

4.用动手深度学习的一个项目测试


1.cuda toolkit Download

cmd输入nvidia-smi,查看电脑支持的版本。



Release Notes :: CUDA Toolkit Documentation

进入上方链接进行比照,选择的cuda版本不能高于你的显卡驱动里面那个版本号,我这边选择了11.2版本。 

从该链接下载对应版本(下载速度有点小慢):CUDA Toolkit Archive | NVIDIA Developer

临时解压路径,默认即可,安装结束后,临时解压文件夹会自动删除。

 同意->自定义,不用选Visual Studio Integration

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
为了安装PyTorchCUDA,您可以按照以下步骤进行操作: 1. 首先,确保您已正确配置好CUDA的环境。CUDA是一种用于进行并行计算的平台和编程模型。您可以从NVIDIA官方网站下载并安装适合您的操作系统的CUDA工具包。 2. 接下来,您需要安装Anaconda,这是一个用于Python的包管理器和环境管理器。您可以从Anaconda官方网站下载适合您的操作系统的Anaconda安装程序,并按照安装向导进行安装。 3. 在安装完Anaconda后,您可以创建一个新的虚拟环境来安装PyTorch。打开命令提示符或终端窗口,运行以下命令: ``` conda create --name myenv ``` 其中"myenv"是您给环境起的名字,您可以根据自己的喜好进行命名。 4. 激活虚拟环境。运行以下命令: - 在Windows系统上: ``` conda activate myenv ``` - 在Mac或Linux系统上: ``` source activate myenv ``` 5. 接下来,您可以使用conda命令安装PyTorch。根据您的操作系统和CUDA版本,您可以选择从清华大学或中国科学技术大学的镜像源下载安装。 - 在Windows系统上,您可以运行以下命令: ``` conda install pytorch torchvision cudatoolkit=<your_cuda_version> -c pytorch ``` - 在Mac或Linux系统上,您可以运行以下命令: ``` conda install pytorch torchvision cudatoolkit=<your_cuda_version> -c pytorch ``` 其中`<your_cuda_version>`是您的CUDA版本号。 注意:在安装PyTorch之前,请确保您已正确设置CUDA环境变量,以便PyTorch能够正确地使用CUDA。 6. 安装完成后,您可以通过导入`torch`模块来验证PyTorch是否成功安装。运行以下Python代码: ``` import torch print(torch.__version__) ``` 如果没有出现错误,并且能够正确输出PyTorch版本号,那么恭喜您,PyTorch已成功安装并可以使用了。 关于PyTorchCUDA安装,以上是一种常见的方法,您也可以根据自己的需求和操作系统选择其他安装方式。希望对您有所帮助!<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [CUDAPytorch安装教程(包含踩坑内容)](https://blog.csdn.net/weixin_48556553/article/details/129852474)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值