配置及安裝pytorch虛擬環境

1 前言

Jupyter Notebook和pycharm是兩個很好用的編程器，我們平時在學習中可以兩個編輯器交換使用，接下來就開始介紹如何配置及安裝虛擬環境，

首先輸入命令

conda create -n 虛擬環境名字 python=版本

例如：我想要創建的虛擬環境名字叫做pytorch，python版本為3.6

conda create -n pytorch python=3.6

在這里插入圖片描述

創建完虛擬環境后，結尾會顯示如何激活虛擬環境
在這里插入圖片描述

我們隨便選一條命令輸入即可

conda activate pytorch

或者

conda deactivate

在這里插入圖片描述
左邊顯示的就是我們環境的名字

在這里插入圖片描述

我們發現我們的虛擬環境中沒有pytorch，所以我們得去安裝
先進去pytorch官網
pytorch官網

在這里插入圖片描述

先查看下面博客的鏈接搭建GPU運算環境
https://blog.csdn.net/weixin_43492572/article/details/106268615
在這里插入圖片描述

我的顯卡支持CUDA11.2及以下版本，
所以我的安裝命令：
在這里插入圖片描述

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

也可以選擇Pip命令
在這里插入圖片描述

pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

輸入pip list命令查看
在這里插入圖片描述

我們輸入python進入python環境
在這里插入圖片描述

然后輸入命令

import torch

沒有報錯的話，再輸入命令檢驗是否可以使用gpu

torch.cuda.is_available()

在這里插入圖片描述

如果像我這樣子的話那就是安裝成功并且也能成功使用gpu，如果電腦上沒有gpu的話那就回傳的是false，

轉載請註明出處，本文鏈接：https://www.uj5u.com/houduan/300250.html

標籤：python