pytorch学习
星辰辰大海
这个作者很懒,什么都没留下…
展开
-
解决pytorch Error loading caffe2_detectron_ops_gpu.dll的错误
报错信息如下:C:\software\Anaconda3\envs\pytorch_190\python.exe C:/Users/stars_ocean/Desktop/PyTorch_test_folder/模型测试.pyTraceback (most recent call last): File "C:/Users/stars_ocean/Desktop/PyTorch_test_folder/timm模型测试.py", line 1, in <module> imp原创 2022-03-29 16:30:26 · 5438 阅读 · 0 评论 -
关于pytorch1.9.0中cpp_extension的默认CUDA版本是11.0的问题
今天在用CUDAExtension编译一个算子的时候,发现它默认的CUDA路径指向的是11.0版本,而我的CUDA版本是11.1.这会导致使用nvcc编译时,会报错No such file or directory错误。这时我们只需要将CUDA的路径指向我们安装的路径即可,比如下面这样torch.utils.cpp_extension.CUDA_HOME = "/usr/local/cuda-11.1"就能够将CUDA路径纠正。...原创 2021-12-07 11:59:48 · 1144 阅读 · 0 评论 -
关于升级pytorch1.9后出现cannot import name ‘container_abcs‘ from ‘torch._six‘错误的解决方法
因为1.8版本之后container_abcs就已经被移除了。cannot import name 'container_abcs' from 'torch._six'在使用一些开源代码的时候,可能会遇到该错误。解决方法也很简单我们只需要给它换一个就行。TORCH_MAJOR = int(torch.__version__.split('.')[0])TORCH_MINOR = int(torch.__version__.split('.')[1])if TORCH_MAJOR ==原创 2021-09-18 09:37:46 · 11351 阅读 · 6 评论 -
解决pytorch的optim.load_state_dict将变量还原到CPU上的问题
最近在学习使用pytorch,遇到一个坑。从中断的训练中继续训练需要重新加载检查点,而有一些优化器(比如adam)的一些变量需要被保存到检查点中。而在使用load_state_dict()还原时,有可能会将这些变量还原到CPU上,如下面的代码这样:model = Model()optimizer = Optim()checkpoint = torch.load(ckp_path)model.load_state_dict(checkpoint["model"])optimizer.lo原创 2021-04-16 15:57:24 · 4065 阅读 · 0 评论