环境安装

环境安装

创建python环境

###创建名为ai的环境
(base) PS C:\Users\dazhi> conda create --name ai python=3.8

###查看创建的环境
(base) PS C:\Users\dazhi> conda env list

###切换到已创建的环境
(base) PS C:\Users\dazhi> conda activate ai
(ai) PS C:\Users\dazhi>

### 打开环境

(ai) PS C:\Users\dazhi> jupyter lab

安装pyTorch

###时间较长
# 注意 要根据自己gpu版本进行安装 不然只能使用cpu版本的
(ai) PS C:\Users\dazhi> conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

验证

(ai) PS C:\Users\dazhi> python
>>> import torch
>>> print(f"CUDA available? {torch.cuda.is_available()}")
CUDA available? True

>>> print(torch.rand(5, 3))
tensor([[0.9133, 0.2088, 0.6461],
        [0.2195, 0.7338, 0.6832],
        [0.6559, 0.7604, 0.8300],
        [0.7688, 0.8725, 0.5349],
        [0.7571, 0.7457, 0.7042]])

安装其他可选包

###深度学习中的辅助函数包

(ai) PS C:\Users\dazhi> pip install d2l

jupyter lab

###安装

conda install -c conda-forge jupyterlab

###此步不安装运行下面打开的编辑器运行会报错

###打开
(ai) PS C:\Users\dazhi> pip install d2l

###排错
###运行时安装过torch但显示没有

解决在环境变量中添加如下代码

###查看路径
import sys
print(sys.path)

C:\\Users\\dazhi\\AppData\\Local\\Programs\\Python\\Python38;C:\\Users\\dazhi\\AppData\\Local\\Programs\\Python\\Python38\\lib\\site-packages

(ai-gpu) PS C:\Users\dazhi> pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu116
# 若不安装下面则打开kupyter后不会显示pytorch
(ai-gpu) PS C:\Users\dazhi> conda install -c conda-forge jupyterlab 


(ai-gpu) PS C:\Users\dazhi> pip3 install d2l

# 若d2l安装后不显示执行上面一步再安装