安装CUDA
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
| ```nvcc --version```命令查看当前CUDA版本。
然后前往[pytorch](https://pytorch.org/get-started/locally/)查看对应版本的pytorch和cuda，NVIDIA官网下载对应版本的cuda安装。
```bash
# Windows下安装
下载安装即可。
安装后，需要添加环境变量CUDA_HOME，和CUDA_PATH的值相同即可。
# Linux下安装
参考官网
# 写入环境变量：vim ~/.bashrc
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
# 保存后：source ~/.bashrc
|
安装torch
如果有安装错误版本的cuda，先将CUDA卸载，并且将pytorch也卸载。
pip uninstall torch torchvision torchaudio
在当前python环境下安装pytorch（官网命令行，CUDA ver12.1）:
1
| pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
|
可用以下命令检查torch是否安装成功：
1
2
3
4
| python
import torch
print(torch.__version__)
print(torch.cuda.is_available())
|