摘要:本文首先在D盤建立了虛擬環境,然後在虛擬環境中安裝了PyTorch,最後配置了pycharm的直譯器。
1. 在 D 盤建立虛擬環境
開啟Anaconda Prompt
輸入conda info --envs
檢視當前已有環境
建立自己的虛擬環境,打算命名為py310,輸入
conda create --prefix=D:\ProgramData\py310 python=3.10
再次檢視虛擬環境,輸入conda info --envs
輸入
conda activate D:\ProgramData\py310
進入虛擬環境
此時,建立的虛擬環境還沒有名字。需要給它取個別名。
輸入conda config --show envs_dirs
檢視一下當前環境目錄。發現沒有新建立的這個環境。
現在要把py310加入envs_dirs
中
conda config --append envs_dirs D:\ProgramData
輸入conda info --envs
,可以看到,此時環境有了別名
嘗試用別名啟用虛擬環境
conda deactivate
先退出,再用別名進入
conda activate py310
成功
2. 在虛擬環境中安裝 PyTorch
我的顯示卡是4060,算力是8.9,CUDA Version是12.6。進入PyTorch官網
只要是CUDA小於等於12.6就行,首選Conda,複製命令
conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia
貼上在Anaconda Prompt中。等待安裝。
安裝好後,輸入conda list
檢視是否存在pytorch
輸入python
,開啟python
輸入
import torch
torch.cuda.is_available()
返回True
,則成功
3. 配置 pycharm
新建專案
新增直譯器
建立
在專案中,新建python指令碼檔案
輸入:
import torch
print(torch.cuda.is_available())
執行一下,顯示True