电商行业建设网站,ui网页设计培训学校,网站建设的开发方法有哪些,高端网站设计目录 1. 问题2. 解决方法2.1 安装torch的CUDA版本2.1.1 查看安装的torch是什么版本2.1.2 下载GPU版本的torch 2.2 安装 NVIDIA CUDA2.2.1 下载CUDA Toolkit2.2.2 安装CUDA Toolkit 最终效果 1. 问题
在Python中#xff0c;使用Pytorch时#xff0c;往往会选择使用CUDA模块使用Pytorch时往往会选择使用CUDA模块但import torch后cuda却无法使用比如
2. 解决方法
2.1 安装torch的CUDA版本
2.1.1 查看安装的torch是什么版本
如果你是默认pip install torch它会默认给你安装cpu版本先输入
pip list查看安装的torch是什么版本如果你的是这样那就不行
需要的torch版本长这样
2.1.2 下载GPU版本的torch
官网下载链接https://pytorch.org/
2.2 安装 NVIDIA CUDA
2.2.1 下载CUDA Toolkit
进入CUDA官网下载CUDA Toolkit https://developer.nvidia.com/cuda-downloads选择合适你的配置
2.2.2 安装CUDA Toolkit
下载好的.exe文件长这样 安装过程如下
最终效果 祝各位成功 还有什么问题请留言讨论
