安装cuda和pytorch

0. 需要是nvidia卡

1. 查看自己的nvidia卡的驱动，更新一下驱动。
2. 了解nvidia卡驱动版本号和cuda版本的兼容性，一般都没什么问题。

1. 安装CUDA

1. 在https://developer.nvidia.com/cuda-toolkit找到对应系统的CUDA Toolkit，下载，并按提示安装（路径最好是默认的，最后选择自动添加环境变量）
2. 在win+R cmd中输入nvcc -V查看cuda是否安装成功，以及安装的版本号
   例如：

   Cuda compilation tools, release 12.6, V12.6.85
   Build cuda_12.6.r12.6/compiler.35059454_0

2.安装pytorch

1. 创建conda环境（因人而异，为了避免和其他环境冲突）
   • 使用管理员身份进入CMD（一般在搜索框中键入cmd，然后鼠标右键进入管理员模式）；
   • 参考conda环境创建；

     创建：conda create -n pytorch python=3.9
     查看环境：conda env list
     激活环境：conda activate pytorch

2. 在https://pytorch.org/get-started/locally/官网找到对应版本的pytorch，尤其需要注意版本号（这里cuda最新只能用12.4啦）

   conda install pytorch torchvision torchaudio pytorch-cuda=12.4 -c pytorch -c nvidia

3. 查看pytorch是否安装成功。

(pytorch) C:\Windows\System32>python
Python 3.9.21
>>> import torch
>>> print(torch.__version__)
2.5.1
>>>