ai环境conda带torch整体迁移。
conda打包好的GPU版torch环境,其实很简单,就是conda装好的torch环境env整体打包,然后到新机器上再解压到env路径。
打开搭建好的环境,找自己路径,我默认的是这个。
cd/root/anaconda3/envs/
然后整个文件夹打包。tar -czvf animatex.tar.gz animatex/
上ubuntu官网下一个跨版本的镜像,wsl导入。安装cuda cudnn conda环境,劳请移步往期视频,这里加速,如果迁移到的新机器有环境就省略这步。
wsl --import U24 E:\wsl\ubuntu E:\wsl/ubuntu-24.04.2-wsl-amd64.gz
然后拷贝到迁移到的机器上,解包到conda的环境路径下
tar -xzvf animatex.tar.gz -C /root/anaconda3/envs/
进入环境运行看看,所有的东西都在,测试torch的cuda和cudnn输出没有问题。
import torch
print("CUDA 可用:", torch.cuda.is_available())
print("CUDA 设备数量:", torch.cuda.device_count())
print("CUDA 当前设备:", torch.cuda.current_device())
print("CUDA 设备名称:", torch.cuda.get_device_name(0))
print("cuDNN 可用:", torch.backends.cudnn.is_available())
print("cuDNN 版本:", torch.backends.cudnn.version())