Philcalab
Philcalab
I wanna know why there are still some errors about version when I use python3.6 pytorch1.2 cuda10.0 in anaconda. for example fatal error: cuda_utils.h: No such file or directory
你好,非常感谢您之前的邮件回复。下面是我的一些疑问: 1.我不太清楚您所说的pointnet2_ops前置包有哪些,我是执行了pip install pointnet2,然后看了下pointnet2的github(https://github.com/erikwijmans/Pointnet2_PyTorch#egg=pointnet2_ops&subdirectory=pointnet2_ops_lib),按照它的需求安装了一些包。之后有执行sh compile.sh,并未显示报错。 2.我的cuda是安装好的,nvcc -V的结果cuda-11.8。另外,我的机器好像无法直接通过 conda install cudatoolkit==11.1安装cuda-11.1,如果不加版本号,默认安装cudatoolkit==11.8。我也很奇怪为什么会报找不到cuda的错误,但是看报错好像找到了我本地安装的cuda-11.8。 3.按照您之前的回复,我尝试运行如下代码检查cuda相关信息 import torch print(torch.__version__) print(torch.cuda.is_available()) print(torch.cuda.device_count()) print(torch.backends.cudnn.version()) print(torch.version.cuda) 控制台输出: /home/user/anaconda3/envs/pudm/bin/python /data/lc/PUDM/check_cuda.py 1.9.0+cu111 /home/user/anaconda3/envs/pudm/lib/python3.9/site-packages/torch/cuda/__init__.py:52: UserWarning: CUDA initialization: CUDA driver initialization failed,...
如何设置多GPU训练
你好,请问在您的代码中是否有设置多GPU训练,因为我在train.py中看到了我这样一行代码  ![Uploading 截图 2025-01-08 19-58-59.png…]() ![Uploading 截图 2025-01-08 19-58-59.png…]()