cuda python torch 虚拟环境配置

pythonSuperman2024-03-06 12:53

以下是Pytorch和CUDA对应的版本

python 复制代码

import torch
print(torch.__version__) 
print(torch.cuda.is_available())
print(torch.empty(3,4,device='cuda'))

python 复制代码

conda uninstall cudatoolkit --force
 
conda uninstall cudnn  --force

在xshell处检查，cuda版本是10.1

本文以cuda10.1为例：

python 复制代码

conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

python 复制代码

conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

在vscode里查看，发现安装的只有10.0

python 复制代码

nvcc -V

由于cuda版本为10.0，Python版本最高只能装到3.6

无论怎么删除Python版本，在（base）（pytorch）等虚拟环境下Python的版本还是3,9，所以干脆新建立了一个虚拟环境

python 复制代码

conda create --name env python=3.6

python 复制代码

python --version

查看版本匹配的torch：

python 复制代码

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch