环境小问题:不同虚拟环境切换不同版本的cuda

创建虚拟环境(好久不创忘记了,搜半天)

conda create -n XXX python=3.8

查看环境列表

conda env list

不同虚拟环境的cuda切换

首先下载自己想要的cuda版本(已有cuda11.6,想要下载cuda11.3)

CUDA Toolkit 11.3 Downloads | NVIDIA Developerhttps://developer.nvidia.com/cuda-11.3.0-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=runfile_local

下载runfile文件

wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run

下载之前应该要安装点依赖和给.run文件赋权限,但我应该之前安装过所以没执行这两条代码也成功了

sudo apt update
sudo apt upgrade

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev 

sudo chmod +x cuda_11.3.0_465.19.01_linux.run

下载后执行安装

sudo sh cuda_11.3.0_465.19.01_linux.run

第一步接受协议accept(页面放大点,不然翻半天找不到)

第二步上下键不要选Driver,再Installer

第三步出现A symlink already exists at /usr/local/cuda. Update to this installation?  我选了no,选yes应该直接覆盖掉上一个版本了

然后安装

现在虚拟环境里还是11.6的

去/usr/local下面可以看到有两个版本的cuda,但是软链接连的 cuda 11.6

激活想要使用11.3的虚拟环境,先删除目前的软链接,再重新链接到11.3上

sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda

再nvcc -V查看:

base环境中也变成了11.3

切换到另一个虚拟环境,cuda还是11.6

———————————————————————————————————————————

按个pytorch3d,现在的环境python 3.8,cuda 11.3, pytorch 1.10.1,手动下载:

https://anaconda.org/pytorch3d/pytorch3d/0.7.4/download/linux-64/pytorch3d-0.7.4-py38_cu113_pyt1101.tar.bz2

conda install pytorch3d-0.7.4-py38_cu113_pyt1101.tar.bz2 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值