簡潔安裝配置在Windows環(huán)境下使用vscode開發(fā)pytorch

使用anaconda安裝pytorch，通過vscode集成環(huán)境開發(fā)pytorch

下載 anaconda

1. 下載網(wǎng)址，選擇對應(yīng)系統(tǒng)的版本 https://repo.anaconda.com/archive/
   windows可以選擇Anaconda3-2024.10-1-Windows-x86_64.exe
2. 檢查 conda 是否安裝成功
   conda --version
3. 升級 conda
   conda update conda
   添加鏡像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/

查看已經(jīng)添加的channels

conda config --get channels

4. 安裝conda完成 配置安裝Python虛擬環(huán)境

conda create --name py311 python=3.11
conda activate py311

5. 使用命令檢查配置環(huán)境

conda env list 

安裝pytorch conda install pytorch

1. 直接執(zhí)行
   安裝pytorch，它默認安裝的是cpu版本的pytorch。

conda install pytorch 

要安裝很多插件，需要等一段時間

配置vs code 的pytorch的開發(fā)環(huán)境

• 安裝如下插件
  

2. 測試pytorch是否正確安裝
   新建一個擴展名為py的文件，代碼如下：

   import torch# 檢查是否可以使用GPU
   device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
   print(f"Using device: {device}")# 創(chuàng)建一個簡單的張量
   x = torch.tensor([1.0, 2.0, 3.0], device=device)
   print(f"Tensor x: {x}")# 創(chuàng)建y張量
   y = torch.tensor([4.0, 5.0, 6.0], device=device)
   print(f"Tensor : {y}")
   # 張量加法
   z = x + y
   print(f"Tensor z (x + y): {z}")# 張量乘法
   w = x * y
   print(f"Tensor w (x * y): {w}")
   

3. 運行py文件，右下角選擇py311環(huán)境
   

4. 輸出如下使用CPU的環(huán)境

Using device: cpu
Tensor x: tensor([1., 2., 3.])
Tensor : tensor([4., 5., 6.])
Tensor z (x + y): tensor([5., 7., 9.])
Tensor w (x * y): tensor([ 4., 10., 18.])