通过conda安装cudatoolikit和cudnn
- 查找cudatoolkit
- 安装cudatoolkit
- 安装cudnn
- 安装cudatoolkit-dev
查找cudatoolkit
conda search cudatoolkit
conda search cudatoolkit-dev
conda search cudnn
安装cudatoolkit
conda install cudatoolkit=11.3 -c /anaconda/pkgs/free/
安装cudnn
conda install cudnn=8.2.1 -c /anaconda/pkgs/free/
安装cudatoolkit-dev
conda install cudatoolkit-dev=11.3 -c conda-forge
解决环境中找不到CUDA_HOME的问题
##测试
python
import torch
print(torch.version) # 查看torch版本号
print() # 编译当前版本的torch使用的cuda版本号
print(.is_available()) # 查看当前cuda是否可用于当前版本的Torch，如果输出True，则表示可用