diff --git a/cudatest.py b/cudatest.py new file mode 100644 index 0000000..08f7e1a --- /dev/null +++ b/cudatest.py @@ -0,0 +1,18 @@ +import torch + +# 获取 PyTorch 版本 +pytorch_version = torch.__version__ +# 获取 CUDA 版本 +cuda_version = torch.version.cuda +# 获取 cuDNN 版本 +cudnn_version = torch.backends.cudnn.version() + +print(f"PyTorch 版本: {pytorch_version}") +print(f"CUDA 版本: {cuda_version}") +print(f"cuDNN 版本: {cudnn_version}") + +# 检查是否可以访问 CUDA +if torch.cuda.is_available(): + print("CUDA is available. GPU name:", torch.cuda.get_device_name(0)) +else: + print("CUDA is not available. Please check your installation.") \ No newline at end of file diff --git a/libs/chatchat-server/chatchat/webui_pages/knowledge_base/knowledge_base.py b/libs/chatchat-server/chatchat/webui_pages/knowledge_base/knowledge_base.py index 5b9492d..be3c457 100644 --- a/libs/chatchat-server/chatchat/webui_pages/knowledge_base/knowledge_base.py +++ b/libs/chatchat-server/chatchat/webui_pages/knowledge_base/knowledge_base.py @@ -397,6 +397,8 @@ def knowledge_base_page(api: ApiRequest, is_lite: bool = None): flex=1, cellEditor="agLargeTextCellEditor", cellEditorPopup=True, + autoWidth=True, + cellEditorParams= { "maxLength": 1000} ) gb.configure_column( "to_del",