cuda 驱动检查
(29条消息) Pytorch1.10安装记录(CUDA11.3)_太阳花的小绿豆的博客-CSDN博客
安装pytorch
Anaconda+Torch(gpu版)极速下载安装方法 - 知乎 (zhihu.com)(27条消息) 手把手教你用anaconda安装pytorch最新版_Alpha Auriage的博客-CSDN博客_anaconda安装pytorchvisdom
windows+anaconda安装visdom - 简书 (jianshu.com)
torchsample安装方法
pip install torchsample==0.1.3 -f https://github.com/ozan-oktay/torchsample/tarball/master#egg=torchsample-0.1.3
报错
运行
python 文件名 -c ./configs/文件名
nn_train.py: error: the following arguments are required:pycharm设置python运行参数 - 简书 (jianshu.com)
报错
(28条消息) No module named ‘cv2‘问题解决方法——亲试好用_花里梦雨的博客-CSDN博客
(29条消息) No module named 'sklearn'_Laura_Wangzx的博客-CSDN博客
Torch not compiled with CUDA enabled
用requirements.txt下载的torch都是cpu版本!!!!!
anaconda创建环境要看是用哪个版本的python不然pytorch装不上,一般是python3.7
3060+11.3CUDA+Pytorch避坑指南 - 知乎 (zhihu.com)
把torch换成对应11.5cuda的可以,用的11.6cuda
内存溢出
运行了一段时间后出现了报错
DefaultCPUAllocator: not enough memory: you tried to allocate 3959520 bytes.