windows CUDA更新(最简单方法)+虚拟环境torch和cuda安装

目录

一、Torch和对应cuda安装

1、查看本身电脑的cuda版本

2、查找对应cuda——torch——python版本

3、安装cuda

4、安装torch

二、window 10 NVIDIA cuda版本更新


一、Torch和对应cuda安装

项目使用torch想要使用GPU运行,但是报错,记录一下解决过程

!!!记住,最好重新建立新的虚拟环境,先安装cuda 再安装torch

conda create -n ccf python=3.7

以免之前的环境其他安装包有副作用影响

1、查看本身电脑的cuda版本

打开NVIDIA控制面板——帮助——系统信息——组件——查看cuda版本

2、查找对应cuda——torch——python版本

也可以通过在anaconda环境下输入命令查看可以安装的cuda版本和对应的安装限制

conda search cudatoolkit --info

3、安装cuda

在项目运行的虚拟环境下输入以下命令

因为第一步我的cuda版本是11.4

那可以安装10.2的,只要选择低于或者等于自身电脑cuda版本的就可以

conda install cudatoolkit=10.2

4、安装torch

这里也试过很多网上教程,有很多版本对应,下载完还是没有正常使用GPU,然后使用下面这个可以

这些是可以选择的torch版本,+cu110代表对应cuda版本是11.0

我之前安装了cuda版本是10.2,所以torch就要对应cu102

pip install torch===1.9.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html

到这就可以正常运行啦

二、window 10 NVIDIA cuda版本更新

我一开始的电脑cuda版本很低,才9.0 就算按照上述方法下载torch版本是1.1.0 但是旧的torch版本很多功能都没有,最好还是提高电脑cuda版本

Please update your GPU driver by downloading and installing a new version 
from the URL: http://www.nvidia.com/Download/index.aspx 
Alternatively, go to:  https://pytorch.org 
to install a PyTorch version that has been compiled with your version

Official Drivers | NVIDIA

所以去到这个网址,选择自己系统的信息,找最新的cuda版本

来到这个页面,你如果填好信息,搜索,那给的是最新的版本,我下载后,不适合本机,可能版本调太高了,所以可以通过点击下面蓝笔圈出的beta and old drivers进入新页面

这里对应信息要填写好,可以通过任务管理器查看GPU版本

然后就下载保存运行,我下载的是

下载运行后,可以再次查看cuda版本,就可以了,成功从9.0升级到11.4

torch版本的选择可以更多,我就用10.2cuda     torch=1.9.0+cu102  大功告成

  • 10
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
虚拟环境安装CUDA的具体步骤如下: 1. 首先,确保已经成功安装CUDA和CuDNN。 2. 在虚拟环境安装PyTorch包,以进行测试。这可以通过运行以下命令来完成: ``` conda install pytorch torchvision cudatoolkit=<cuda版本号> -c pytorch ``` 其中,`<cuda版本号>`应替换为您想要安装CUDA版本号。 3. 等待PyTorch包的安装过程完成。 4. 安装完成后,您可以在虚拟环境中使用PyTorch来测试CUDA是否成功安装。您可以编写一个简单的代码来检查CUDA是否可用,例如: ``` import torch print(torch.cuda.is_available()) ``` 如果输出为`True`,则表示CUDA已成功安装并可用。 通过按照上述步骤,在虚拟环境安装CUDA并使用PyTorch进行测试,您将能够确认CUDA是否在虚拟环境中正确安装并配置成功。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *3* [conda虚拟环境安装cuda和cudnn,再也不用头疼版本号的问题了](https://blog.csdn.net/qq_43705697/article/details/121618276)[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_1"}}] [.reference_item style="max-width: 50%"] - *2* [完美解决(最简单):不同虚拟环境不同CUDA版本安装问题(tensorflow-gpu与pytorch)](https://blog.csdn.net/m0_61102347/article/details/120922915)[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_1"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值