pytorch官方网站,cuda官方网站
CUDA下载:https://developer.nvidia.com/cuda-toolkit-archive
CUDNN下载:https://developer.nvidia.com/rdp/cudnn-download
pytorch下载:pytorch.org
任务管理器中只显示CUDA占用的专用内存,而几乎不占用共享内存?
这是win10显示的bug,其实你的gpu使用率已经上去了,但是任务管理器没有正确显示。可以按下win+g键调用微软提供的游戏录制工具查看正确的使用率。
cuda11.5安装指南
https://blog.csdn.net/qq_45281807/article/details/121294644
注意,该博主安装的是cpu版本,所以可能会报错false, 这篇博客的替换包可以解决问题-清华源conda
安装gpu版本的pytorch总是推送cpu版本解决办法
测试CUDA是否安装成功代码
import torch
print(torch.__version__)
print(torch.cuda.is_available())
pychram 里面是扫描不到anaconda的虚拟环境包的,扫描的是python的包
在虚拟环境里面 conda insatll XX;pip insatll XX;
注意pytorch版本对应cuda
cuda环境没有cv2模块
在虚拟环境中,conda insatll opencv-python,或者pip insatll opencv-python。不要conda insatll opencv可能会出现失败