如何在D盤建立虛擬環境?包括安裝PyTorch和配置PyCharm

茴香豆的茴發表於2024-08-31

摘要:本文首先在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

相關文章