Anaconda配置虚拟环境、安装pytorch、cuda笔记

一、Anaconda配置虚拟环境

打开Anaconda Prompt命令行

创建虚拟环境命令如下：

查看已经创建的所有虚拟环境：conda env list  

创建新虚拟环境： conda create -n test python=3.7 #-n 后面加虚拟环境名称，指定python的版本

启动虚拟环境：conda activate test

此时，虚拟环境已经创建完成，接下来在虚拟环境中安装pytorch。

！！！pytorch安装很容易出错，要与cuda、cudnn版本对应！！！

二、安装pytorch、CUDA、cudnn

1.先查看本机Nvidia适用的CUDA版本

打开本机命令行cmd，输入命令：nvidia-smi

红框说明本机最高适用的CUDA版本为11.6，接下来开始安装。

2.安装pytorch

 进入pytorch官网，默认是2.0版本的，不要选。

点击下面的其他版本，因为本机最高支持CUDA11.6，所以小于11.6即可。

复制CUDA11.6这行命令，粘贴到Anaconda Prompt刚才的命令行后

安装完成后，验证pytorch是否安装成功。

命令如下：

python

import torch

print(torch.__version__)

验证CUDA是否安装成功：print(torch.cuda.is_available())

3.安装CUDA

进入CUDA官网：CUDA Toolkit 12.2 Update 1 Downloads | NVIDIA Developer

 

 安装完成后，打开cmd，输入nvcc -V查看CUDA版本号：

至此，虚拟环境配置、pytorch、cuda安装完成。