没有网站想做个链接页面怎么做,江苏省建设厅官网网站首页,保定网站制作,体检中心 网站建设方案目录
一、Torch和对应cuda安装
1、查看本身电脑的cuda版本
2、查找对应cuda——torch——python版本
3、安装cuda
4、安装torch
二、window 10 NVIDIA cuda版本更新 一、Torch和对应cuda安装
项目使用torch想要使用GPU运行#xff0c;但是报错#xff0c;记录一下解决…目录
一、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 python3.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 cudatoolkit10.2
4、安装torch
这里也试过很多网上教程有很多版本对应下载完还是没有正常使用GPU然后使用下面这个可以 这些是可以选择的torch版本cu110代表对应cuda版本是11.0
我之前安装了cuda版本是10.2所以torch就要对应cu102
pip install torch1.9.0cu102 -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 torch1.9.0cu102 大功告成
