1.检查 CUDA 版本:
打开命令行(在 Windows 上是 CMD 或 PowerShell,在 Linux 或 macOS 上是终端),然后输入以下命令:
nvcc --version
nvcc 是 NVIDIA CUDA 编译器,这个命令会显示 CUDA 的版本信息,如果安装正确,你应该能看到版本号和其他相关信息。
2.查看安装的包列表
pip list
或者,查看特定包的信息,可以使用:
pip show <package_name>
将 <package_name> 替换为你想要查看的包的名称。
3.查看python版本
python --version
安装时如果是GPU版本,一定要特别注意是不是需要的!!!
4.查看下载的通道
conda config --show channels
清华大学镜像源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
中科大源:
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/
5. 查看一下自己的虚拟环境
conda list
conda info --envs
6.查看torchvision 和 torchaudio 的版本
python
import torchvision
import torchaudio
print("TorchVision Version:", torchvision.__version__)
print("TorchAudio Version:", torchaudio.__version__)
7.查看CUDA Version版本
nvidia-smi
//安装Pytorch(2.0.1+cu118)
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url 网页链接
8.验证Pytorch是否安装成功
python
import torch
print(torch.__version__)//返回版本号
torch.cuda.is_available()//返回true
9. 查看 torchvision、torchaudio 和 PyTorch(通常指的是 torch 库)的版本
查看 PyTorch 版本:
python
import torch
print(torch.__version__)
查看 torchvision 版本:
python
import torchvision
print(torchvision.__version__)
查看 torchaudio 版本:
python
import torchaudio
print(torchaudio.__version__)
查看 PyTorch CUDA 版本: 如果你想知道 PyTorch 是否支持 CUDA 以及支持哪个版本的 CUDA,可以使用以下命令:
python
print(torch.version.cuda)