安装显卡驱动、安装cuda toolkit、安装cudnn 把这三个文件夹的文件分别拷贝到CUDA安装目录对应的(bin、include、lib)文件夹中即可。CUDA的lib目录有x64 、Win32、cmake三个文件夹,拷到其中的x64这个文件夹中。参考:https://blog.csdn.net/ZHUO__zhuo/article/details/130536374。如果 你前面没有安装vs,直接安装的这个,需要把CUDA里面的。为下载cudnn创建了NVIDIA账号记录:有账号就不管。出现了你安装的CUDA的版本信息,说明安装成功了。
pyqt安装 并导入pychram pip install PyQt5 -i https://pypi.douban.com/simplepip install PyQt5 -i https://pypi.douban.com/simpleE:\anaconda3\envs\torch\Lib\site-packages\qt5_applications\Qt\binE:\anaconda3\envs\torch\Scripts\pyuic5.exe
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the 解决:self.dataLoader = DataLoader(VOCDataset(),batch_size=8,shuffle=True,num_workers=1)原因:代码只把数据放入cuda了,而没有把网络放入cuda。原因:批次太大,根据自己硬件情况 调整最大的批次。解决:把网络放入cuda;
Pycharm切换新环境 Pycharm切换anaconda的环境 【Pycharm切换新环境 Pycharm切换anaconda的环境】(打开Pycharm的File->setting)(点击python interpreter)