pytorch无法使用显卡运行程序的解决办法 |
您所在的位置：网站首页 › 检测不到显卡驱动怎么办 › pytorch无法使用显卡运行程序的解决办法 |
pytorch无法使用显卡运行程序的解决办法
|
WIN10系统正确安装对应版本cudatoolkit-10.1.243、cudnn-7.6.5、cuda-10.1.243、cudnn补丁后仍然无法使用显卡运行程序的解决办法
测试Cuda版本测试Anaconda版本并创建环境安装pytorch-gpu
测试Cuda版本
win+R打开运行窗口，输入cmd打开命令窗口 输入命令 >nvcc -V显示如下结果，证明cuda已经安装成功 nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2019 NVIDIA Corporation Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019 Cuda compilation tools, release 10.1, V10.1.243 测试Anaconda版本并创建环境在cmd窗口输入指令 >conda --version返回版本号，证明安装成功 打开Anaconda窗口，选择Environment，点击creat创建自己需要的python版本，我创建的是3.7版本，环境名称自己取一个好记的 前往Pytorch官网，按照自己需要的配置进行选择，第一项可以默认选择稳定版，选择完成后复制生成的下载命令。 在cmd窗口激活自己的环境，放入刚才粘贴的代码，等待安装 >activate （你的环境名称） >conda install pytorch torchvision cudatoolkit=10.1 -c pytorch安装完成后测试显卡能否使用 >python Python 3.7.7 (default, May 6 2020, 11:45:54) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32 Type "help", "copyright", "credits" or "license" for more information. >>>import torch >>>torch.cuda.is_available() True显示True证明成功 |
【本文地址】
今日新闻 |
推荐新闻 |