-
Notifications
You must be signed in to change notification settings - Fork 148
Description
请问以下我在配置Lane detection 的数据集culane的demo中,当训练启动python main_landet.py --test --config=configs/lane_detection/bezierlanenet/resnet34_culane_aug1b.py --checkpoint=resnet34_bezierlanenet_culane_aug1b_20211109.pt
遇到问题
NVIDIA GeForce RTX 3060 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the NVIDIA GeForce RTX 3060 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
报错
我配的环境安装官网配的pytorch1.6 nvidia-smi显示cuda是12.1, nvcc -V,显示没有,请问环境方面有什么问题吗
下面是环境的
(pad) cxf@cxf:~$ conda list
packages in environment at /home/cxf/anaconda3/envs/pad:
Name Version Build Channel
_libgcc_mutex 0.1 main
_openmp_mutex 5.1 1_gnu
absl-py 1.4.0 pypi_0 pypi
addict 2.4.0 pypi_0 pypi
blas 1.0 mkl
ca-certificates 2023.05.30 h06a4308_0
cachetools 4.2.4 pypi_0 pypi
certifi 2021.5.30 py36h06a4308_0
charset-normalizer 2.0.12 pypi_0 pypi
cudatoolkit 10.2.89 hfd86e86_1
dataclasses 0.8 pypi_0 pypi
dill 0.3.4 pypi_0 pypi
filetype 1.0.8 pypi_0 pypi
freetype 2.12.1 h4a9f257_0
future 0.18.3 pypi_0 pypi
google-auth 2.20.0 pypi_0 pypi
google-auth-oauthlib 0.4.6 pypi_0 pypi
grpcio 1.48.2 pypi_0 pypi
idna 3.4 pypi_0 pypi
imageio 2.10.1 pypi_0 pypi
importlib-metadata 4.8.3 pypi_0 pypi
importmagician 0.1.0 pypi_0 pypi
intel-openmp 2022.1.0 h9e868ea_3769
joblib 1.1.1 pypi_0 pypi
jpeg 9e h5eee18b_1
lcms2 2.12 h3be6417_0
ld_impl_linux-64 2.38 h1181459_1
lerc 3.0 h295c915_0
libdeflate 1.17 h5eee18b_0
libffi 3.3 he6710b0_2
libgcc-ng 11.2.0 h1234567_1
libgomp 11.2.0 h1234567_1
libpng 1.6.39 h5eee18b_0
libstdcxx-ng 11.2.0 h1234567_1
libtiff 4.5.0 h6a678d5_2
libwebp-base 1.2.4 h5eee18b_1
lz4-c 1.9.4 h6a678d5_0
markdown 3.3.7 pypi_0 pypi
mkl 2020.2 256
mkl-service 2.3.0 py36he8ac12f_0
mkl_fft 1.3.0 py36h54f3939_0
mkl_random 1.1.1 py36h0573a6f_0
mmcv-full 1.3.5 pypi_0 pypi
multiprocess 0.70.12.2 pypi_0 pypi
ncurses 6.4 h6a678d5_0
ninja 1.11.1 pypi_0 pypi
ninja-base 1.10.2 hd09550d_5
numpy 1.19.2 py36h54aff64_0
numpy-base 1.19.2 py36hfa32c7d_0
oauthlib 3.2.2 pypi_0 pypi
olefile 0.46 py36_0
opencv-python 4.5.4.58 pypi_0 pypi
openjpeg 2.4.0 h3ad879b_0
openssl 1.1.1t h7f8727e_0
p-tqdm 1.3.3 pypi_0 pypi
pathos 0.2.8 pypi_0 pypi
pillow 8.4.0 pypi_0 pypi
pip 21.2.2 py36h06a4308_0
pox 0.3.0 pypi_0 pypi
ppft 1.6.6.4 pypi_0 pypi
protobuf 3.19.6 pypi_0 pypi
pyasn1 0.5.0 pypi_0 pypi
pyasn1-modules 0.3.0 pypi_0 pypi
python 3.6.13 h12debd9_1
pytorch 1.6.0 py3.6_cuda10.2.89_cudnn7.6.5_0 pytorch
pyyaml 6.0 pypi_0 pypi
readline 8.2 h5eee18b_0
requests 2.27.1 pypi_0 pypi
requests-oauthlib 1.3.1 pypi_0 pypi
rsa 4.9 pypi_0 pypi
scikit-learn 0.23.2 pypi_0 pypi
scipy 1.5.4 pypi_0 pypi
setuptools 58.0.4 py36h06a4308_0
shapely 1.8.0 pypi_0 pypi
six 1.16.0 pyhd3eb1b0_1
sqlite 3.41.2 h5eee18b_0
tensorboard 2.7.0 pypi_0 pypi
tensorboard-data-server 0.6.1 pypi_0 pypi
tensorboard-plugin-wit 1.8.1 pypi_0 pypi
thop 0.0.31-2005241907 pypi_0 pypi
threadpoolctl 3.1.0 pypi_0 pypi
timm 0.4.5 pypi_0 pypi
tk 8.6.12 h1ccaba5_0
torchvision 0.7.0 py36_cu102 pytorch
tqdm 4.62.3 pypi_0 pypi
typing-extensions 4.1.1 pypi_0 pypi
ujson 4.2.0 pypi_0 pypi
urllib3 1.26.16 pypi_0 pypi
werkzeug 2.0.3 pypi_0 pypi
wheel 0.37.1 pyhd3eb1b0_0
xz 5.4.2 h5eee18b_0
yapf 0.32.0 pypi_0 pypi
zipp 3.6.0 pypi_0 pypi
zlib 1.2.13 h5eee18b_0
zstd 1.5.5 hc292b87_0