共计 646 个字符,预计需要花费 2 分钟才能阅读完成。
簡單記錄一下人在大陸要如何絲滑配置本地大模型開發環境,畢竟很多境外網站都要掛代理才能訪問,特別是拉取 Python 包,雖然有清華鏡像源,但配置 Cuda 這部分還是有點技巧的。
配置 CUDA
首先打開命令窗口後輸入以下命令查看 CUDA 版本,
ndvia-smi
nvcc --version
如下圖
如果你沒安裝應該是空的,請自己到 Nvdia 網站 下載安裝
配置 Torch
打 PyTorch 網站 根據版本選擇對應的命令
如果默認不選擇,都是直接使用 CPU,運行 AI 環境 CPU 跑起來會很多問題,甚至無法運行,所以建議使用 CUDA
這個時候打開 PyCharm 或者任何命令窗口,如下圖執行安裝命令,你會發現真的很慢!
這時候把網址複製出來打開迅雷,直接滿速下載!差距 10 倍啊?♂️
下載完成後改從本地 whl 包中獲取依賴包,如下命令
最後進入 Python 窗口確認是否配置成功,如下圖
import torch print(torch.cuda.is_available()) print(torch.__version__)
順帶一提測試之前記得先 import torch
正文完





