pytorch GPU版本安装CUDA11.6

由于之前下了cpu版本,所以又重新下了GPU版本

首先查看CUDA版本
cmd中输入 NVIDIA-smi
在这里插入图片描述
再打开Anaconda Prompt
1,创建环境
conda create -n Pytorch-env python==3.7
2.激活环境
conda activate Pytorch-env
3.pytorch安装
输入 pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
(可以在网站中找对应自己版本的命令
https://pytorch.org/get-started/locally/#no-cuda-1)
pytorch GPU版本安装CUDA11.6_第1张图片
4.安装成功后,运行以下命令检验是否下载的gpu版本的pytorch
python
import torch
torch.cuda.is_available()

在这里插入图片描述
显示True安装成功

5.下载新环境的spyder
conda install spyder==4.1.3
为确保安装好所有的依赖再输入:
pip install spyder==4.1.3

额外:
由于一些命令需要用到sklearn包 新环境下没有 所以下载sklearn包
conda install numpy
conda install pandas
conda install scipy
conda install matplotlib
conda install scikit-learn==0.20.0

你可能感兴趣的:(pytorch,深度学习,python)